Exploring the frontiers of softmax: Provable optimization, applications in diffusion model, and beyond (arXiv 2024)