Prateek Jain

Principal Scientist, Google DeepMind
Adjunct Faculty, IIT Kanpur
Contact: Google DeepMind, No. 3
RMZ Infinity Tower E, Old Madras Road
Bangalore 560 016 India

I am a Principal Scientist / Diretor 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.

Brief Bio for Talks


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