Software Engineer

Highlights

• Design distributed data pipelines powering large-scale ML training workloads.
• Improve training goodput by optimizing scalable, reliable input systems.
• Help advance the Waymo Driver behind 10M+ autonomous trips.

What they offer

• Annual bonus program
• Equity incentive plan
• Company benefits program

Ideal candidate

An experienced ML infrastructure engineer who thrives on large-scale distributed systems and partnering with researchers to unblock training performance and improve reliability at scale.

Requirements

• Bachelor's degree in Computer Science, Mathematics, or related field and 5 years progressive, post-baccalaureate experience
• 5 years of experience in machine learning frameworks and distributed data processing libraries, including TensorFlow, JAX, ML Pathways, Apache Beam
• 5 years of experience in Python or C++ for machine learning libraries and distributed systems
• 2 years of experience in technical leadership in architecting ML infrastructure for large foundation models (LLMs)
• 2 years of experience in productionizing, evaluating, serving large foundation models in multi-host distributed ML runtimes
• 2 years of experience in optimizing computational efficiency of distributed ML training or inference