Research Engineer, Infrastructure and Scale

AI Breaking Wire; Mountain View, Northern (CA, KY); Hybrid; USD 165,000 - 230,000; Full time

Benefits offered by this job

Stock options

Healthcare coverage

PTO

Retirement match

Job summary

Google DeepMind is seeking a Research Engineer for the Infrastructure and Scale team. You will build foundational systems that train state‑of‑the‑art AI models and scale computation across large TPU clusters.

Responsibilities include designing high‑throughput distributed training frameworks, profiling performance bottlenecks on GPUs/TPUs, and collaborating with researchers to co‑design new features and system capabilities.

Qualifications

• Strong systems background with distributed computing experience.
• Proficiency in C++ and Python is required.
• Experience with MPI, JAX or TensorFlow is preferred.

Responsibilities

• Architect and optimize high-throughput distributed training frameworks for large models.
• Profile and debug performance bottlenecks across hardware accelerators and software stacks.
• Collaborate with researchers to co-design new algorithmic features and system capabilities.

Skills

Strong systems background

C++

Python

Distributed systems

Education

Bachelor's degree or higher in CS/EE or related

Tools

MPI

JAX

TensorFlow

Job description

About the Role

Google DeepMind is looking for a Research Engineer to join our core Infrastructure and Scale team. You will build the foundational systems that train some of the world's most advanced AI models. Your work will directly impact how we scale computation across massive TPU clusters, driving the next wave of scientific and algorithmic breakthroughs.

Responsibilities

• Architect and optimize high-throughput distributed training frameworks for large language models and multimodal systems.
• Profile and debug complex performance bottlenecks across hardware accelerators, interconnects, and software stacks.
• Collaborate with researchers to co-design new algorithmic features and system-level capabilities.

Requirements

• B.S., M.S., or Ph.D. in Computer Science, Electrical Engineering, or a related technical discipline.
• Strong systems background with proficiency in C++ and Python.
• Hands-on experience with distributed systems, MPI, JAX, or TensorFlow.
• Experience scaling workloads across large clusters of GPUs or TPUs.

Benefits

• Competitive compensation package including salary, bonus, and stock units.
• Comprehensive healthcare and wellness programs.
• Generous paid time off and retirement matching programs.
• Access to cutting-edge computing resources and continuous learning opportunities.