Research Engineer, Gemini Code Post-training, DeepMind

Google; Town of Montana (WI); On-site; USD 174,000 - 252,000; Full time

Job summary

Google DeepMind seeks a Research Engineer to advance Gemini code post-training using reinforcement learning and supervised fine-tuning for large language models. You will develop scalable post-training pipelines, collaborate with Operations Research teams to benchmark performance, and design automated evaluation suites to ensure robust improvements.

You will contribute to infrastructure, reward models, and data curation for iterative model releases while engaging with a global research ecosystem

Qualifications

• Bachelor's degree in CS/ML/AI or equivalent practical experience.
• 5 years of experience developing ML models with JAX, PyTorch, or TensorFlow.
• Experience with post-training RL or supervised fine-tuning for LLMs.
• Experience designing or running LLM evaluation benchmarks and pipelines.

Responsibilities

• Drive post-training research and engineering using RL and SFT to advance Gemini coding capabilities.
• Develop and scale post-training pipelines and benchmarks with OR teams to set industry-leading performance.
• Design, build, and maintain evaluation suites and automated benchmarks.
• Implement training infrastructure, reward models, and data curation workflows.

Skills

ML research

Frameworks: JAX/PyTorch/TensorFlow

Post-training RL / SFT for LLMs

LLM evaluation benchmarks

Education

Bachelor's degree in CS/ML/AI or equivalent

Tools

TPU/GPUs acceleration

Job description

Research Engineer, Gemini Code Post-training, DeepMind

Location:

Mountain View, CA, USA

Minimum qualifications:

• Bachelor's degree in Computer Science, Machine Learning, Artificial Intelligence, or a related technical field, or equivalent practical experience.
• 5 years of experience in software engineering or research developing machine learning models with frameworks such as JAX, PyTorch, or TensorFlow.
• Experience conducting post-training, reinforcement learning (RL), or supervised fine-tuning (SFT) for large language models (LLMs).
• Experience designing or running LLM evaluation benchmarks and measurement pipelines.

Preferred qualifications:

• Experience in agentic coding workflows, code generation, or software engineering tasks across web, mobile, 3D, or game development.
• Experience with competitive coding benchmarks, Code Arena, or public model leaderboards.
• Experience scaling distributed training pipelines on TPU or GPU accelerators.
• Experience with reward modeling, preference optimization, or synthetic data generation for code models.
• Experience working in fast-paced research environments delivering iterative model releases.

About the job

At Google, research-focused Software Engineers are embedded throughout the company, allowing them to set up large-scale tests and deploy promising ideas quickly and broadly. Ideas may come from internal projects as well as from collaborations with research programs at partner universities and technical institutes all over the world.

From creating experiments and prototyping implementations to designing new architectures, engineers work on real-world problems including artificial intelligence, data mining, natural language processing, hardware and software performance analysis, improving compilers for mobile platforms, as well as core search and much more. But you stay connected to your research roots as an active contributor to the wider research community by partnering with universities and publishing papers.

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: $174,000 - $252,000 (USD) + 15% bonus target + equity + benefits

Learn more about benefits at Google.

Responsibilities

• Drive post-training research and engineering using reinforcement learning (RL) and supervised fine-tuning (SFT) to advance Gemini coding capabilities across web, 3D, game, and mobile development.
• Develop and scale agentic post-training pipelines and landing recipes in collaboration with Operations Research (OR) teams to establish industry-leading benchmark performance in Code Arena.
• Design, build, and maintain frontier evaluation suites and automated benchmarks to measure, stress-test, and improve agentic coding capabilities.
• Implement training infrastructure, reward models, and data curation workflows to accelerate iterative model improvements from revision to revision.

Information collected and processed as part of your Google Careers profile, and any job applications you choose to submit is subject to Google's Applicant and Candidate Privacy Policy.

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.

If you have a need that requires accommodation, please let us know by completing our Accommodations for Applicants form.

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.

To all recruitment agencies: Google does not accept agency resumes. Please do not forward resumes to our jobs alias, Google employees, or any other organization location. Google is not responsible for any fees related to unsolicited resumes.

Equity is granted exclusively and discretionally 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.

Locations:

Mountain View, CA, USA