Medapati, Sourabh

5 publications

ICLR 2025 Accelerating Neural Network Training: An Analysis of the AlgoPerf Competition Priya Kasimbeg, Frank Schneider, Runa Eschenhagen, Juhan Bae, Chandramouli Shama Sastry, Mark Saroufim, Boyuan Feng, Less Wright, Edward Z. Yang, Zachary Nado, Sourabh Medapati, Philipp Hennig, Michael Rabbat, George E. Dahl

TMLR 2025 How Far Away Are Truly Hyperparameter-Free Learning Algorithms? Priya Kasimbeg, Vincent Roulet, Naman Agarwal, Sourabh Medapati, Fabian Pedregosa, Atish Agarwala, George E. Dahl

ICML 2025 Towards Flexible Perception with Visual Memory Robert Geirhos, Priyank Jaini, Austin Stone, Sourabh Medapati, Xi Yi, George Toderici, Abhijit Ogale, Jonathon Shlens

NeurIPS 2024 Amortized Planning with Large-Scale Transformers: A Case Study on Chess Anian Ruoss, Grégoire Delétang, Sourabh Medapati, Jordi Grau-Moya, Li Kevin Wenliang, Elliot Catt, John Reid, Cannada A. Lewis, Joel Veness, Tim Genewein

NeurIPSW 2023 Adaptive Gradient Methods at the Edge of Stability Jeremy Cohen, Behrooz Ghorbani, Shankar Krishnan, Naman Agarwal, Sourabh Medapati, Michal Badura, Daniel Suo, Zachary Nado, George E. Dahl, Justin Gilmer