Learning Triton One Kernel at a Time: Softmax | Towards Data Science

All you need to know about a fast, readable and PyTorch-ready softmax kernel!