Research Scientist, Agent Post Training, DeepMind

On-site$174,000 – $253,000
United States

Tech Stack

TRl (Transformer Reinforcement Learning)FlaxJAXMachine LearningPyTorch

Job Description, Responsibilities & Requirements

About the Position

Join Google DeepMind as a Research Scientist, Agent Post Training. Drive AI research from hypothesis to delivery in the Gemini model recipe.

Responsibilities

  • Drive the research process for large-scale agent post-training from hypothesis formulation to delivery in the Gemini model recipe.
  • Design and execute ablation studies to validate research hypotheses and accelerate experimental feedback loops.
  • Communicate research findings, progress, and outcomes to the broader team through visualizations and reports.
  • Develop research infrastructure and utilities for data analysis and model evaluations using standard engineering practices.
  • Collaborate with other research scientists and engineers to maintain a regular feedback and communication loop.

Requirements

  • PhD in Computer Science, Machine Learning, or a related quantitative field, or equivalent practical experience.
  • 2 years of experience with machine learning frameworks such as JAX, Flax, or PyTorch.
  • Experience conducting research in reinforcement learning, tool use, or agentic systems.

Nice to Have

  • Experience publishing research in reinforcement learning, reinforcement learning from human feedback, or tool-use algorithms at machine learning venues.
  • Experience working with large-scale distributed training infrastructure and scaling experiments.
  • Research experience in systems design, code complexity, or working with large-codebase environments.
  • Experience developing simple, scalable solutions for complex, open-ended research problems.

We Offer

  • Competitive salary: $174,000 - $253,000 (USD) + 15% bonus target + equity + benefits.
  • Opportunity to work in a pioneering AI lab with exceptional interdisciplinary teams.
  • Diverse learning opportunities and varied career pathways.

About the Company

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.

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.

Job Details

Company name:
FOR RENT APP
Salary:
$174,000 – $253,000
Location:
United States
Employment Type:
Full-time
Work Mode:
On-site
Posted on TheJob:
Jul 18, 2026
Last checked:
Jul 18, 2026
Apply Now
© 2026 TheJob, Inc. All rights reserved.