← All roles

Staff Research Scientist, Gemini Research, DeepMind

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

About the role

Staff Research Scientist role at DeepMind, Mountain View, CA, focusing on inventing and validating new ML algorithms and training systems for Gemini research, with responsibilities spanning algorithm design, automated research systems, and production-facing improvements.

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.
  • Experience working with deep learning frameworks (e.g., JAX/Flax, PyTorch) for distributed systems and accelerator clusters.
  • One or more scientific publication submission(s) for conferences, journals, or public repositories.

Skills

Required:Machine learning algorithm designDistributed systems for MLJAX/Flax, PyTorchCode optimization on TPUsResearch publication

Benefits

  • Learn more about benefits at Google (link provided in description).
  • Equity is granted exclusively and discretionarily by Alphabet Inc. on the basis of an agreement concluded between you and Alphabet Inc.
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