Scholar
Arnav Kumar Jain
Google Scholar ID: tu7wKckAAAAJ
Université de Montréal/ Mila
Citations & Impact
Citations
2,205
H-index
8
i10-index
7
Publications
15
Co-authors
14
list available
Publications
Resume (English only)
Academic Achievements
- Paper 'A Smooth Sea Never Made a Skilled SAILOR: Robust Imitation via Learning to Search' accepted at NeurIPS 2025 (Spotlight); Paper 'Multi-Turn Code Generation Through Single-Step Rewards' accepted at ICML 2025 (Spotlight).
Research Experience
- Worked as a Data & Applied Scientist at Microsoft; Interned at Cohere with Matthias Gallé on LLM agents for code generation.
Education
- Final year Ph.D. student at Université de Montréal and Mila, advised by Prof. Irina Rish, and closely collaborating with Sanjiban Choudhury. Earned an Integrated M.Sc. in Mathematics and Computing from Indian Institute of Technology Kharagpur, where he spent time in KRSSG on building autonomous soccer playing robots.
Background
- Excited about developing agents that interact intelligently with the world to learn how to search, adapt, and recover from mistakes. Before starting his PhD, he was a Data & Applied Scientist at Microsoft, collaborating with Dr. Manik Varma on eXtreme Classification for web-scale recommender systems.
Miscellany
- Currently on the job market for industry research scientist roles.
