🚀 Featured Publications

NIPS'24

Mixture of Nested Experts: Adaptive Processing of Visual Tokens

Gagan Jain, Nidhi Hegde, ...

NeurIPS 2024Mixture of ExpertsConditional Computation

TL;DR: Token-wise routing between nested experts for tackling redundancy.

ECCV'24

LookupViT: Compressing Visual Information to a smaller number of tokens

Rajat Koner, Gagan Jain, ...

ECCV 2024VisionEfficiency

TL;DR: An asynchronous version of attention with sub-quadratic scaling and superior performance.

📰 News & Updates Archive

[Dec'25]🎤Speaking at WADLA 4.0 at IIITDM Jabalpur.
[Dec'25]🎉Organising Scaling Post Training for LLMs workshop at ICLR 2026.
[May'25]📄Decode time scaling accepted at ICML'25 & ICLR'25 Workshop.
[Dec'24]📄Collaborative bandits accepted at AAMAS'25! Optimal regret bounds!
[Dec'24]🎤Presented efficient inference at UCLA Artificial General Intelligence Lab.

Last updated: June 2025