pip install maskedtensor
You need to run a recent nightly of PyTorch. You can get it on the main website.
For an introduction and instructions on how to use MaskedTensors and what they are useful for see the notebooks folder.
maskedtensor is licensed under BSD 3-Clause