k-Mixup Regularization for Deep Learning via Optimal Transport. - Kristjan H. Greenewald; Anming Gu; Mikhail Yurochkin; Justin Solomon 001; Edward Chien (Trans. Mach. Learn. Res., 2023)

Compute-Optimal LLMs Provably Generalize Better with Scale. - Marc Anton Finzi; Sanyam Kapoor; Diego Granziol; Anming Gu; Christopher De Sa; J. Zico Kolter; Andrew Gordon Wilson (ICLR, 2025)

Partially Observed Trajectory Inference using Optimal Transport and a Dynamics Prior. - Anming Gu; Edward Chien; Kristjan H. Greenewald (ICLR, 2025)

Mirror Mean-Field Langevin Dynamics. - Anming Gu; Juno Kim (CoRR, 2025)

Private Continuous-Time Synthetic Trajectory Generation via Mean-Field Langevin Dynamics. - Anming Gu; Edward Chien; Kristjan H. Greenewald (CoRR, 2025)