Publications

Building Customer Support AI Agents at 100M-User Scale: An Evaluation-Driven Framework

Aman Gupta, Kevin Rossell, Edesio Alcobaça, José Chrystian Lima Pacheco, Carolina Baptista de Lima, Shao Tang, Luiz Paulo Rabachini, Luis Moneda, Herbert Fei, Daniel Silva, Rohan Ramanath

arxivdoi

Effective Quantization of Muon Optimizer States

Aman Gupta, Rafael Celente, Abhishek Shivanna, DT Braithwaite, Gregory Dexter, Shao Tang, Hiroto Udagawa, Daniel Silva, Rohan Ramanath, S Sathiya Keerthi

Preprint posted 2025

arxiv

Building Production LLM Agents: An Evaluation-Driven Playbook

Aman Gupta

Tutorials TrackTo appear

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, Jason Zhu, Natesh Pillai, S Sathiya Keerthi

Implemented in Hugging Face TRL

arxivdoi

Efficient Algorithms for Leveraging LLMs for Generative and Predictive Recommender Systems

Kayhan Behdin, Yun Dai, Gregory Dexter, Aman Gupta, Rahul Mazumder, Ankan Saha, Qingquan Song, Shao Tang, Sirou Zhu, Pin-Lun Hsu

Tutorials Trackcorresponding author, joint 1st author

doi

From Features to Transformers: Redefining Ranking for Scalable Impact

Fedor Borisyuk, Lars Hertel, Ganesh Parameswaran, Gaurav Srivastava, Sudarshan Srinivasa Ramanujam, Borja Ocejo, Peng Du, Aman Gupta, and 9 others

Withdrawn by the authors; no PDF available

arxiv

LANTERN: Scalable Distillation of Large Language Models for Job-Person Fit and Explanation

Zhoutong Fu, Yihan Cao, Yi-Lin Chen, Aman Lunia, Liming Dong, Neha Saraf, Ruijie Jiang, Aman Gupta, and 11 others

arxiv

LLM Query Scheduling with Prefix Reuse and Latency Constraints

Gregory Dexter, Shao Tang, Ata Fatahi Baarzi, Qingquan Song, Tejas Dharamsi, Aman Gupta

arxivdoi

On the Robustness of Reward Models for Language Model Alignment

Jiwoo Hong, Noah Lee, Eunki Kim, Guijin Son, Woojin Chung, Aman Gupta, Shao Tang, James Thorne

arxivdoi

Scaling Down, Serving Fast: Compressing and Deploying Efficient LLMs for Recommendation Systems

Kayhan Behdin, Ata Fatahibaarzi, Qingquan Song, Yun Dai, Aman Gupta, Zhipeng Wang, Shao Tang, Hejian Sang, and 12 others

joint 1st author

arxiv

Your Spending Needs Attention: Modeling Financial Habits with Transformers

DT Braithwaite, Misael Cavalcanti, R Austin McEver, Hiroto Udagawa, Daniel Silva, Rohan Ramanath, Felipe Meneses, Arissa Yoshida, Evan Wingert, Matheus Ramos, Brian Zanfelice, Aman Gupta

arxiv

360Brew: A Decoder-only Foundation Model for Personalized Ranking and Recommendation

Hamed Firooz, Maziar Sanjabi, Adrian Englhardt, Aman Gupta, Ben Levine, Dre Olgiati, Gungor Polatkan, Iuliia Melnychuk, and 16 others

Withdrawn by the authors; no PDF available

arxiv

A Precise Characterization of SGD Stability Using Loss Surface Geometry

Gregory Dexter, Borja Ocejo, Sathiya Keerthi, Aman Gupta, Ayan Acharya, Rajiv Khanna

arxivdoi

Efficient user history modeling with amortized inference for deep learning recommendation models

Lars Hertel, Neil Daftary, Fedor Borisyuk, Aman Gupta, Rahul Mazumder

arxiv

FFSplit: Split Feed-Forward Network For Optimizing Accuracy-Efficiency Trade-off in Language Model Inference

Zirui Liu, Qingquan Song, Qiang Charles Xiao, Sathiya Keerthi Selvaraj, Rahul Mazumder, Aman Gupta, Xia Hu

arxiv

LiMAML: Personalization of Deep Recommender Models via Meta Learning

Ruofan Wang, Prakruthi Prabhakar, Gaurav Srivastava, Tianqi Wang, Zeinab S Jalali, Varun Bharill, Yunbo Ouyang, Aman Gupta, and 5 others

arxivdoi

LiNR: Model Based Neural Retrieval on GPUs at LinkedIn

Fedor Borisyuk, Qingquan Song, Mingzhou Zhou, Ganesh Parameswaran, Madhu Arun, Siva Popuri, Tugrul Bingol, Aman Gupta, and 6 others

arxivdoi

LiRank: Industrial Large Scale Ranking Models at LinkedIn

Fedor Borisyuk, Mingzhou Zhou, Qingquan Song, Siyu Zhu, Birjodh Tiwana, Ganesh Parameswaran, Siddharth Dangi, Aman Gupta, and 26 others

arxivdoi

MultiSlot ReRanker: A Generic Model-based Re-Ranking Framework in Recommendation Systems

Qiang Charles Xiao, Ajith Muralidharan, Birjodh Tiwana, Johnson Jia, Fedor Borisyuk, Aman Gupta, Dawn Woodard

arxiv

Neural Optimization with Adaptive Heuristics for Intelligent Marketing System

Changshuai Wei, Benjamin Zelditch, Joyce Chen, Andre Assuncao Silva T Ribeiro, Jingyi Kenneth Tay, Borja Ocejo Elizondo, Sathiya Keerthi Selvaraj, Aman Gupta, Licurgo Benemann De Almeida

arxivdoi

mSAM: Micro-Batch-Averaged Sharpness-Aware Minimization

Kayhan Behdin, Qingquan Song, Aman Gupta, Sathiya Keerthi, Ayan Acharya, Borja Ocejo, Gregory Dexter, Rajiv Khanna, David Durfee, Rahul Mazumder

arxiv

Optimizing for Member Value in an Edge Building Marketplace

Ayan Acharya, Siyuan Gao, Ankan Saha, Borja Ocejo, Kinjal Basu, Keerthi Selvaraj, Rahul Mazumder, Aman Gupta, Parag Agrawal

doi

Practical Design of Performant Recommender Systems using Large-scale Linear Programming-based Global Inference

Aman Gupta, S Sathiya Keerthi, Ayan Acharya, Miao Cheng, Borja Ocejo Elizondo, Rohan Ramanath, Rahul Mazumder, Kinjal Basu, J Kenneth Tay, Rupesh Gupta

Tutorials Track

doicodesite

Promoting Inactive Members in Edge-Building Marketplace

Ayan Acharya, Siyuan Gao, Borja Ocejo, Kinjal Basu, Ankan Saha, Keerthi Selvaraj, Rahul Mazumdar, Parag Agrawal, Aman Gupta

Companion Proceedings

doi

QuantEase: Optimization-based Quantization for Language Models

Kayhan Behdin, Ayan Acharya, Aman Gupta, Qingquan Song, Siyu Zhu, Sathiya Keerthi, Rahul Mazumder

arxiv

A Light-speed Linear Program Solver for Personalized Recommendation with Diversity Constraints

Haoyue Wang, Miao Cheng, Kinjal Basu, Aman Gupta, Keerthi Selvaraj, Rahul Mazumder

arxiv

Generalized Deep Mixed Models

Jun Shi, Chengming Jiang, Aman Gupta, Mingzhou Zhou, Yunbo Ouyang, Qiang Charles Xiao, Qingquan Song, Yi Wu, Haichao Wei, Huiji Gao

doi

Heterogeneous Calibration: A post-hoc model-agnostic framework for improved generalization

David Durfee, Aman Gupta, Kinjal Basu

arxiv

Improved Deep Neural Network Generalization Using m-Sharpness-Aware Minimization

Kayhan Behdin, Qingquan Song, Aman Gupta, David Durfee, Ayan Acharya, Sathiya Keerthi, Rahul Mazumder

arxiv

Adam vs. SGD: Closing the generalization gap on image classification

Aman Gupta, Rohan Ramanath, Jun Shi, S Sathiya Keerthi

pdf

Logit Attenuating Weight Normalization

Aman Gupta, Rohan Ramanath, Jun Shi, Anika Ramachandran, Sirou Zhou, Mingzhou Zhou, S Sathiya Keerthi

arxiv

Image and video understanding for recommendation and spam detection systems

Aman Gupta, Sirjan Kafle, Di Wen, Dylan Wang, Sumit Srivastava, Suhit Sinha, Nikita Gupta, Bharat Jain, Ananth Sankar, Liang Zhang

doi

Smoothed Gaussian Mixture Models for Video Classification and Recommendation

Aman Gupta, Rohan Ramanath, Jun Shi, Anika Ramachandran, Sirou Zhou, Mingzhou Zhou, S Sathiya Keerthi

arxiv

Extracting compact representation of knowledge from gene expression data for protein-protein interaction

Haohan Wang, Aman Gupta, Ming Xu

doi

Sparse Network Inference using the k-Support Norm

Aman Gupta, Haohan Wang, Rama Kumar Pasumarthi

pdf

Learning structure in gene expression data using deep architectures, with an application to gene clustering

Aman Gupta, Haohan Wang, Madhavi Ganapathiraju

doipreprint

Mycobacterium tuberculosis and Clostridium difficille interactomes: demonstration of rapid development of computational system for bacterial interactome prediction

Seshan Ananthasubramanian, Rahul Metri, Ankur Khetan, Aman Gupta, Adam Handen, Nagasuma Chandra, Madhavi Ganapathiraju

doi