hierarchicalsoftmax (pypi): A Hierarchical Softmax Framework for PyTorch.
Keywords: pytorch|softmax