Drimva

Waymo

Software Engineer

$250K–$263K

PERM - N/A +1 · Full-time · Posted Oct 21, 2026

Location
PERM - N/A
Mountain View, California, United States
Experience
5 years of progressive, post-baccalaureate experience in the job offered or a Software Engineering-related occupation; stated technology and specialty experience periods are listed in requirements.
Education
Bachelor's degree in Computer Science, Mathematics, or a related field.

About the job

Designs and improves distributed data pipelines and ML infrastructure to make large-scale model training more scalable, reliable, and efficient.

What you'll do

  • Design and improve distributed input data pipelines for large-scale machine learning training workloads.
  • Collaborate with researchers and ML engineers to resolve pipeline bottlenecks and improve training goodput.
  • Implement and maintain ML infrastructure tools, evaluate technologies, and promote distributed-systems best practices.

What they ask for

  • Bachelor's degree in Computer Science, Mathematics, or a related field.
  • 5 years of experience with machine learning frameworks and distributed data processing libraries, and 5 years using Python or C++ for machine learning libraries and distributed systems.
  • 2 years of technical leadership architecting ML infrastructure for large foundation models (LLMs).
  • 2 years of experience productionizing, evaluating, and serving large foundation models in multi-host distributed ML runtimes; 2 years optimizing computational efficiency of distributed ML training or inference.

Pay

$250,000 - $263,000 USD (base salary)

Tools

TensorFlow, JAX, ML Pathways, Grain, Apache Beam, Python, C++

AI focus

Builds and optimizes distributed ML infrastructure for training, evaluating, and serving large foundation models (LLMs).

Model-written summary · the original posting is authoritative

First seen Oct 22, 2026 · Last verified Nov 4, 2026 (UTC)