Aman Gupta

Researcher, Nubank

Joined

May 2021

Names

Aman Gupta (Preferred)

Emails

****@linkedin.com (Confirmed)

****@gmail.com (Confirmed)

****@nubank.com.br (Confirmed)

Personal Links

Google Scholar

DBLP

Career & Education History

Researcher

Nubank (nubank.com.br)

2025 – Present

Researcher

LinkedIn (linkedin.com)

2019 – 2024

Senior Machine Learning Scientist

Apple (apple.com)

2016 – 2018

MS student

School of Computer Science, Carnegie Mellon University (cs.cmu.edu)

2014 – 2016

Advisors, Relations & Conflicts

No relations added

Expertise

deep learning

optimization

Present

linear programming

Present

recommender systems

explore-exploit

Present

Publications

On quantizing the state of the Muon optimizer
Aman Gupta, Rafael Celente, Abhishek Shivanna, Daniel Thomas Braithwaite, Gregory Dexter, Shao Tang, Hiroto Udagawa, Daniel Silva, Rohan Ramanath, Sathiya Keerthi
- SPOT
- Readers: Everyone
In-Context Bayesian Reward Modeling for Test-Time Steerability
Jiwoo Hong, Shao Tang, Zhipeng Wang, Aman Gupta
- ICLR 2026 Conference Withdrawn Submission
- Readers: Everyone
On quantizing the state of the Muon optimizer
Aman Gupta, Shao Tang, Gregory Dexter, Abhishek Shivanna, Rafael Celente, Rohan Ramanath, Daniel Silva, Hiroto Udagawa, Daniel Thomas Braithwaite, Sathiya Keerthi
- Submitted to ICLR 2026
- Readers: Everyone
LLM Query Scheduling with Prefix Reuse and Latency Constraints
Gregory Dexter, Shao Tang, Ata Fatahibaarzi, Qingquan Song, Tejas Dharamsi, Aman Gupta
- NeurIPS 2025 poster
- Readers: Everyone
Adam vs. SGD: Closing the generalization gap on image classification
Aman Gupta
- OPT
- Readers: Everyone
LiRank: Industrial Large Scale Ranking Models at LinkedIn
Fedor Borisyuk, Aman Gupta
- KDD
- Readers: Everyone
LLM Query Scheduling with Prefix Reuse and Latency Constraints
Gregory Dexter, Shao Tang, Ata Fatahi Baarzi, Qingquan Song, Tejas Dharamsi, Aman Gupta
- CoRR 2025
- Readers: Everyone
Efficient AI in Practice: Training and Deployment of Efficient LLMs for Industry Applications
Kayhan Behdin, Yun Dai, Ata Fatahibaarzi, Aman Gupta, Qingquan Song, Shao Tang, Hejian Sang, Gregory Dexter, Sirou Zhu, Siyu Zhu, Tejas Dharamsi, Maziar Sanjabi, Vignesh Kothapalli, Hamed Firooz, Zhoutong Fu, Yihan Cao, Pin-Lun Hsu, Fedor Borisyuk, Zhipeng Wang, Rahul Mazumder et al. (2 additional authors not shown)
- CoRR 2025
- Readers: Everyone
From Features to Transformers: Redefining Ranking for Scalable Impact
Fedor Borisyuk, Lars Hertel, Ganesh Parameswaran, Gaurav Srivastava, Sudarshan Srinivasa Ramanujam, Borja Ocejo, Peng Du, Andrei Akterskii, Neil Daftary, Shao Tang, Daqi Sun, Qiang Charles Xiao, Deepesh Nathani, Mohit Kothari, Yun Dai, Aman Gupta
- CoRR 2025
- Readers: Everyone
AlphaPO: Reward Shape Matters for LLM Alignment
Aman Gupta, Shao Tang, Qingquan Song, Sirou Zhu, Jiwoo Hong, Ankan Saha, Viral Gupta, Noah Lee, Eunki Kim, Siyu Zhu, Parag Agrawal, Natesh S. Pillai, Sathiya Keerthi
- ICML 2025 poster
- Readers: Everyone

View all 44 publications

Co-Authors

View all 146 co-authors