Combining Axes Preconditioners through Kronecker Approximation for Deep Learning - Duvvuri, Venkata Sai Surya Subramanyam; Devvrit, Fnu; Anil, Rohan; Hsieh, Cho-Jui; Dhillon, Inderjit (International Conference on Learning Representations, 2024)