Research Engineer: Large-Scale Pretraining & ML Systems

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

Benefits offered by this job

Equity

Benefits

Bonus target

Job summary

Google DeepMind in Mountain View, CA seeks a research-focused Software Engineer to advance large-scale ML systems and research initiatives. You will design, prototype, and deploy software enabling cutting-edge AI work, collaborate with cross-disciplinary teams, and contribute to performance and scalability of pretraining pipelines.

The role emphasizes strong Python software engineering, ML knowledge, and the ability to work across time zones with global partners, while maintaining safety and

Qualifications

• Bachelor's degree in CS or related technical field or equivalent practical experience.
• 2 years of software engineering experience with Python-based ML/scientific libraries (e.g., JAX).
• 1 year of experience with machine learning.

Responsibilities

• Work on long term projects enabling new research with multiple team members.
• Respond to time-sensitive issues arising in large-scale training runs.
• Improve technical workflows and tooling used across pretraining.
• Apply engineering expertise to help researchers navigate technical challenges.
• Balance robust engineering practices with a flexible, researcher-friendly codebase.

Skills

Python

Machine Learning

Software Engineering

Education

Bachelor's degree in CS or related field

Tools

JAX

Job description

Google DeepMind in Mountain View, CA seeks a research-focused Software Engineer to advance large-scale ML systems and research initiatives. You will design, prototype, and deploy software enabling cutting-edge AI work, collaborate with cross-disciplinary teams, and contribute to performance and scalability of pretraining pipelines.

The role emphasizes strong Python software engineering, ML knowledge, and the ability to work across time zones with global partners, while maintaining safety and