Proximal
MTS, Research
Building reinforcement learning environments and post-training models.
02 / Gagan Jain
Background
Experience & education
Full CV ↗September 2026 — present
May 2025 — August 2026
Microsoft AI
Applied Scientist II · Bengaluru
Large-vocabulary generative retrieval under latency constraints.
September 2023 — May 2025
Google DeepMind / Google Research
Pre-Doctoral Researcher · Bengaluru
Training and decoding efficiency, token compression, conditional computation, and low-rank bandits.
July 2022 — September 2023
Microsoft R&D
Data & Applied Scientist · Hyderabad
Extreme classification, dense retrieval, and recommender systems.
2018 — 2022
Indian Institute of Technology Bombay
B.Tech. (Hons.), Mechanical Engineering
Minors in AI & Data Science and Computer Science & Engineering.
CPI: 9.33/10 · Department rank: 4/126
Early research experience
December 2018 — April 2022
Undergraduate Researcher · IIT Bombay
Research in autonomous driving, visual perception, and control, advised by Amit Sethi and Shivaram Kalyanakrishnan.
Selected honors
2022
Technical Citation & Organizational Color
Indian Institute of Technology Bombay
2021
IARC world champion team
International Aerial Robotics Competition
2019
ASME SDC Asia-Pacific champion team
Student Design Competition
Beyond research
Away from research, I enjoy hiking, badminton, tennis, traveling, and mystery thrillers.