Softmax September 23, 2024 updated October 27, 2024 1 min read [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 ml