Prateek Jain


Distinguished Scientist, Google DeepMind
Adjunct Faculty, IIT Kanpur
Contact: Google DeepMind,
Google Ananta, Mahadevpura
Bangalore 560 037 India
Email: prajain@google.com



I am a Distinguished Scientist at Google DeepMind India where I co-lead long-term model research for Gemini as well as 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. I have been fortunate to lead/contribute to exciting projects like Matformers, MRL, Tandem, Treeformer, HIRE, Provable Non-convex Optimization, Stochastic Optimization.
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.

Brief Bio for Talks




News

27 February 2024

Talk at IISc Workshop on RL, 2024

15 December 2023

Matformers received best paper award at Neurips ENLSP, 2023

10-16 December 2023

Nine results from MLO presented at Neurips, 2023

Oct 3 2023

Presented Matformers at ICCV, 2023

25 July 2023

Four results from MLO presented at ICML, 2023

13-16 July 2023

Co-organizing COLT in Bangalore. See you there!

7 July 2023

Talk on "Learning with Dependent Data" at ICTS Workshop on Data Science -- Probabilistic and Optimization Methods

26 March 2022

Talk on "Online learning with Markovian data via reverse experience replay" at Ashoka University Workshop on Learning and Data Science

6 December 2021

Received ACM India Early Career Researcher Award

... see all News