Control-flow refinement and progress invariants for bound analysis. - Sumit Gulwani; Sagar Jain; Eric Koskinen (PLDI, 2009)

Factorized Recurrent Neural Architectures for Longer Range Dependence. - Francois Belletti; Alex Beutel; Sagar Jain; Ed Huai-hsin Chi (AISTATS, 2018)

Practical Diversified Recommendations on YouTube with Determinantal Point Processes. - Mark Wilhelm; Ajith Ramanathan; Alexander Bonomo; Sagar Jain; Ed H. Chi; Jennifer Gillenwater (CIKM, 2018)

Q&R: A Two-Stage Approach toward Interactive Recommendation. - Konstantina Christakopoulou; Alex Beutel; Rui Li 049; Sagar Jain; Ed H. Chi (KDD, 2018)

Categorical-attributes-based item classification for recommender systems. - Qian Zhao; Jilin Chen; Minmin Chen; Sagar Jain; Alex Beutel; Francois Belletti; Ed H. Chi (RecSys, 2018)

Latent Cross: Making Use of Context in Recurrent Recommender Systems. - Alex Beutel; Paul Covington; Sagar Jain; Can Xu; Jia Li; Vince Gatto; Ed H. Chi (WSDM, 2018)

Seq2Slate: Re-ranking and Slate Optimization with RNNs. - Irwan Bello; Sayali Kulkarni; Sagar Jain; Craig Boutilier; Ed Huai-hsin Chi; Elad Eban; Xiyang Luo; Alan Mackey; Ofer Meshi (CoRR, 2018)

Top-K Off-Policy Correction for a REINFORCE Recommender System. - Minmin Chen; Alex Beutel; Paul Covington; Sagar Jain; Francois Belletti; Ed H. Chi (WSDM, 2019)

Towards Neural Mixture Recommender for Long Range Dependent User Sequences. - Jiaxi Tang; Francois Belletti; Sagar Jain; Minmin Chen; Alex Beutel; Can Xu; Ed H. Chi (WWW, 2019)

Understanding and Improving Knowledge Distillation. - Jiaxi Tang; Rakesh Shivanna; Zhe Zhao 001; Dong Lin; Anima Singh; Ed H. Chi; Sagar Jain (CoRR, 2020)

DCN-M: Improved Deep & Cross Network for Feature Cross Learning in Web-scale Learning to Rank Systems. - Ruoxi Wang; Rakesh Shivanna; Derek Zhiyuan Cheng; Sagar Jain; Dong Lin; Lichan Hong; Ed H. Chi (CoRR, 2020)

DCN V2: Improved Deep & Cross Network and Practical Lessons for Web-scale Learning to Rank Systems. - Ruoxi Wang; Rakesh Shivanna; Derek Zhiyuan Cheng; Sagar Jain; Dong Lin; Lichan Hong; Ed H. Chi (WWW, 2021)

Clock Aligned Input Adiabatic Logic. - Aditya S. Kumar; Sagar Jain; Neeta Pandey (Microelectron. J., 2021)

Multi-Resolution Attention for Personalized Item Search. - Furkan Kocayusufoglu; Tao Wu; Anima Singh; Georgios Roumpos; Heng-Tze Cheng; Sagar Jain; Ed H. Chi; Ambuj K. Singh (WSDM, 2022)

Algorithms for Efficiently Learning Low-Rank Neural Networks. - Kiran Vodrahalli; Rakesh Shivanna; Maheswaran Sathiamoorthy; Sagar Jain; Ed H. Chi (CoRR, 2022)

Density Weighting for Multi-Interest Personalized Recommendation. - Nikhil Mehta 002; Anima Singh; Xinyang Yi; Sagar Jain; Lichan Hong; Ed H. Chi (CoRR, 2023)

A Mortarboard-Shaped Receiver Antenna for Drone Wireless Charging to Achieve an Outspread Misalignment Tolerance Towards Imperfect Landing. - Sagar Jain; Ananth Bharadwaj; Ashwani Sharma (IEEE Trans. Veh. Technol., 2025)

DV365: Extremely Long User History Modeling at Instagram. - Wenhan Lyu; Devashish Tyagi; Yihang Yang; Ziwei Li; Ajay Somani; Karthikeyan Shanmugasundaram; Nikola Andrejevic; Ferdi Adeputra; Curtis Zeng; Arun K. Singh; Maxime Ransan; Sagar Jain (CoRR, 2025)