Is it possible to add support for sequence mask?

[qiyifei1]

It would be useful to mask out some of the inputs if possible.

[rish-16]

Hello, definitely second this! Would love to have support for masking if it isn't too much trouble. Using this repo for a research project and it's great so far :)

Thank you, Phil!