LookupViT: Compressing Visual Information to a Limited Number of Tokens. - Rajat Koner; Gagan Jain; Prateek Jain 002; Volker Tresp; Sujoy Paul (ECCV (86), 2024)

Mixture of Nested Experts: Adaptive Processing of Visual Tokens. - Gagan Jain; Nidhi Hegde 003; Aditya Kusupati; Arsha Nagrani; Shyamal Buch; Prateek Jain 002; Anurag Arnab; Sujoy Paul (NeurIPS, 2024)

Bayesian Collaborative Bandits with Thompson Sampling for Improved Outreach in Maternal Health Program. - Arpan Dasgupta; Gagan Jain; Arun Sai Suggala; Karthikeyan Shanmugam; Milind Tambe; Aparna Taneja (CoRR, 2024)

Bayesian Collaborative Bandits with Thompson Sampling for Improved Outreach in Maternal Health. - Arpan Dasgupta; Gagan Jain; Arun Sai Suggala; Karthikeyan Shanmugam; Milind Tambe; Aparna Taneja (AAMAS, 2025)

Masked Generative Nested Transformers with Decode Time Scaling. - Sahil Goyal; Debapriya Tula; Gagan Jain; Pradeep Shenoy; Prateek Jain 002; Sujoy Paul (CoRR, 2025)