← All roles

Staff Research Scientist, Gemini Research, DeepMind

Google· DeepMind
Apply on company site
Mountain View, CA, USA$207k - $300kfull-timePosted 1w ago

About the role

Researcher role focused on inventing ML architectures, optimizers, and training methods for Gemini; acts as final technical verifier, building scalable ML systems and contributing to production models within the Gemini team at DeepMind, Google.

Responsibilities

  • Invent, implement, and scale new Machine Learning (ML) algorithms, such as architectures, optimizers, training objectives, and recipes.
  • Design and build automated research systems in which Gemini proposes, codes, and evaluates changes to its own training stack, moving toward a model that designs its successor.
  • Serve as the final verifier for AI-proposed algorithm changes: reproduce results, design decisive ablations across model scales, rule out confounds such as evaluation leakage or unequal tuning budgets, and decide what moves forward.
  • Build and own lean, high-performance training and evaluation code (JAX on Tensor Processing Units, with custom kernels where needed) allowing researchers and AI agents can test ideas at meaningful scale.
  • Partner with Gemini pre-training, post-training, and infrastructure teams to land validated improvements in production models, and share results through internal talks.

Qualifications

  • PhD in Computer Science, a related field, or equivalent practical experience.
  • 4 years of experience with research agendas across multiple teams or projects.
  • One or more scientific publication submission(s) for conferences, journals, or public repositories.
  • Experience working with deep learning frameworks (e.g., JAX/Flax, PyTorch) for distributed systems and accelerator clusters.

Skills

Required:Machine Learning researchDistributed systemsProgramming with JAX/Flax or PyTorch

Benefits

  • Learn more about benefits at Google (link provided in description).
Last updated Oct 23, 2026. Interested? Apply on the company site.

About Google

Google Search homepage

llmtext-to-speechvoice-ai-stack
Company profileWebsite

More open roles at Google