Staff Research Scientist, Gemini Research, DeepMind

DeepMind Technologies Limited; Mountain View (CA); On-site; USD 207,000 - 300,000; Full time

Job summary

DeepMind Technologies Limited in Mountain View, CA, seeks a senior researcher to invent ML architectures, optimizers, and training methods, and to build the systems that let Gemini propose, implement, and test such changes on its own. You will be the team's final technical verifier, deciding which AI-generated ideas are real advances and carrying the winners into next-generation Gemini models.

This is a deep individual-contributor role.

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 ML algorithms, architectures, optimizers, training objectives, and recipes.
• Design and build automated research systems enabling Gemini to propose, code, and evaluate changes to its training stack.
• Serve as final verifier for AI-proposed algorithm changes: reproduce results and rule out confounds.
• Build and own lean, high-performance training and evaluation code (JAX on TPUs, with custom kernels).
• Partner with Gemini teams to land improvements in production models and share results.

Skills

Deep learning

Research experience

Publications

Distributed systems

Education

PhD in CS or related field

Equivalent practical experience

Tools

JAX

Flax

PyTorch

Job description

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

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

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.

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

Responsibilities

Learn more about benefits at Google.

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

Google is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. See also Google's EEO Policy and EEO is the Law. If you have a disability or special need that requires accommodation, please let us know by completing our Accommodations for Applicants form.