Cohen, Jeremy

7 publications

ICLR 2025 Understanding Optimization in Deep Learning with Central Flows Jeremy Cohen, Alex Damian, Ameet Talwalkar, J Zico Kolter, Jason D. Lee

NeurIPS 2025 Understanding the Evolution of the Neural Tangent Kernel at the Edge of Stability Kaiqi Jiang, Jeremy Cohen, Yuanzhi Li

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

NeurIPSW 2022 On the Maximum Hessian Eigenvalue and Generalization Simran Kaur, Jeremy Cohen, Zachary Chase Lipton

ICLR 2021 Gradient Descent on Neural Networks Typically Occurs at the Edge of Stability Jeremy Cohen, Simran Kaur, Yuanzhi Li, J Zico Kolter, Ameet Talwalkar

ICML 2021 Pareto GAN: Extending the Representational Power of GANs to Heavy-Tailed Distributions Todd Huster, Jeremy Cohen, Zinan Lin, Kevin Chan, Charles Kamhoua, Nandi O. Leslie, Cho-Yu Jason Chiang, Vyas Sekar

ICML 2019 Certified Adversarial Robustness via Randomized Smoothing Jeremy Cohen, Elan Rosenfeld, Zico Kolter