Masked Softmax Layers in PyTorch
mcognetta.github.io