Principal Scientist, Google DeepMind
Adjunct Faculty, IIT Kanpur
Contact:
Google DeepMind, No. 3
RMZ Infinity Tower E, Old Madras Road
Bangalore 560 016 India
Email: prajain@google.com
I am a Principal Scientist / Director at Google DeepMind India where I also lead the Machine Learning and Optimization team. My research interests are in machine learning, efficient and elastic inference of large models, retrieval and indexing, rigorous non-convex optimization and reinforcement learning.
Earlier, I was a Sr. Principal Research Scientist at Microsoft Research India. I completed my PhD at the University of Texas at Austin under Prof. Inderjit S. Dhillon.
Talk at IISc Workshop on RL, 2024
15 December 2023Matformers received best paper award at Neurips ENLSP, 2023
10-16 December 2023Nine results from MLO presented at Neurips, 2023
Oct 3 2023Presented Matformers at ICCV, 2023
25 July 2023Four results from MLO presented at ICML, 2023
13-16 July 2023Co-organizing COLT in Bangalore. See you there!
7 July 2023Talk on "Learning with Dependent Data" at ICTS Workshop on Data Science -- Probabilistic and Optimization Methods
26 March 2022Talk on "Online learning with Markovian data via reverse experience replay" at Ashoka University Workshop on Learning and Data Science
6 December 2021Received ACM India Early Career Researcher Award