[2410.01104] softmax is not enough (for sharp out-of-distribution) How to write a fast Softmax kernel - YouTube Log Softmax Log Exp Sum logsumexp Online Softmax Hierarchical Softmax