Adaptive Gradient Methods at the Edge of Stability. - Jeremy Cohen 001; Behrooz Ghorbani; Shankar Krishnan; Naman Agarwal; Sourabh Medapati; Michal Badura; Daniel Suo; David Cardoze; Zachary Nado; George E. Dahl; Justin Gilmer (CoRR, 2022)

Benchmarking Neural Network Training Algorithms. - George E. Dahl; Frank Schneider 001; Zachary Nado; Naman Agarwal; Chandramouli Shama Sastry; Philipp Hennig; Sourabh Medapati; Runa Eschenhagen; Priya Kasimbeg; Daniel Suo; Juhan Bae; Justin Gilmer; et al. (CoRR, 2023)