Masked Softmax Layers in PyTorch | Not Hacker News!