Staff Research Scientist, Gemini Research, DeepMind

Google LLC; Mountain View (CA); On-site; USD 207,000 - 300,000; Full time

Benefits offered by this job

Equity

Annual bonus

Job summary

Google DeepMind in Mountain View, CA, seeks a Staff Research Scientist for Gemini Research. You will invent ML architectures, optimizers, and training methods, and build systems to test ideas at scale across distributed GPUs/TPUs.

This is a deep individual-contributor role requiring a PhD or equivalent, strong publication record, and leadership across teams. You will work with JAX/Flax and PyTorch, collaborating with pre-training, post-training, and infrastructure teams to land improvements into

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.

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.

Skills

Research leadership

Deep learning frameworks

Publications

Distributed systems

Education

PhD in Computer Science or related field

Tools

JAX/Flax

PyTorch

Job description

Staff Research Scientist, Gemini Research, DeepMind

Share Staff Research Scientist, Gemini Research, DeepMind

corporate_fare DeepMind place Mountain View, CA, USA

• 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.

Preferred qualifications:

• 2 years of experience in coding and leading multiple research efforts and influencing research direction.

About the job

Just as grading olympiad proofs takes someone who can solve olympiad problems, judging an AI-proposed change to a frontier training algorithm takes a researcher who could have invented it. In this role, you are that researcher.

You will invent new machine learning (ML) architectures, optimizers, and training methods, and build the systems that let Gemini propose, implement, and test such changes on its own. As the team's final technical verifier, you will decide which AI-generated ideas are real advances and carry the winners into next-generation Gemini models.

This is a deep individual-contributor role. You'll combine original research in ML algorithm and architecture design with engineering across the full stack, from model code and accelerator kernels to distributed training and evaluation.

Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high-quality product innovation for billions of users. We use our technologies for widespread public benefit and scientific discovery, ensuring safety and ethics are always our highest priority.

We are pushing the boundaries across multiple domains. Our global teams offer diverse learning opportunities and varied career pathways for those driven to achieve exceptional results through collective effort. Individual pay is determined by factors including job-related skills, experience, and relevant education or training.

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.

US: $207000 - $300000 (USD) + 20% bonus target + equity + benefits

Learn more about benefits at Google .

Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law. See also Google's EEO Policy , Know your rights: workplace discrimination is illegal , Belonging at Google , and How we hire .

Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles unless stated otherwise in the job posting.

Equity is granted exclusively and discretionarily by Alphabet Inc. on the basis of an agreement concluded between you and Alphabet Inc. Alphabet Inc. is your sole contractual partner with respect to equity grants. GSU grants are not guaranteed, are discretionary, are subject to approval by the Alphabet Inc. board of directors or its delegate, the terms of the relevant Alphabet Inc. stock plan, and your grant agreement. They have no impact on statutory payments. Current or past grants do not confer an acquired right.