SOAP: Improving and Stabilizing Shampoo using Adam for Language Modeling - Vyas, Nikhil; Morwani, Depen; Zhao, Rosie; Shapira, Itai; Brandfonbrener, David; Janson, Lucas; Kakade, Sham (International Conference on Learning Representations, 2025)