Google DeepMind · Bangalore

Research Scientist, Machine Learning Optimization

28.10.2025

Description

  • Design, implement and evaluate models, agents and software prototypes of large foundational models.
  • Deep dive into fundamentals of both the ML aspects of foundational models (like architectures, loss functions, data, evals) as well as their implementation on neural accelerators (efficiency during training, serving). 
  • Report and present research findings and developments including status and results clearly and efficiently both internally and externally, verbally and in writing.
  • Suggest and engage in team collaborations to meet ambitious research goals.
  • Work with external collaborators and maintain relationships with relevant research labs and key individuals as appropriate.
  • Work in collaboration with our Responsible AI teams to ensure our advances in intelligence are developed ethically and provide broad benefits to humanity.

Qualifications

  • PhD in a technical field or equivalent practical experience.
  • PhD in Machine Learning, Computer Vision, Natural Language Processing or related areas
  • A proven track record of publications and relevant experience in one or more areas such as large foundational models, ML+Systems, optimization, learning theory, computer vision, NLP etc.
  • Proven experience with ML frameworks (e.g. JAX)  and proven experience with training large models 
  • Proven experience working in industry, working on projects from proof-of-concept through to implementation, applying experimental ideas to applied problems
  • A real passion for AI, Optimization, and Efficiency!

Application

View listing at origin and apply!