Waymo
Senior Software Engineer, Training Efficiency
$238K–$302K
Mountain View, California, United States · Senior · Full-time · Posted Aug 11, 2025
- Experience
- 5+ years of equivalent real-world experience as an alternative to a B.S.
- Education
- B.S. in Computer Science or Math, or 5+ years of equivalent real-world experience.
About the job
Improves the runtime efficiency, scalability, and reliability of distributed input data pipelines for large-scale machine learning training.
What you'll do
- Design and improve distributed input data pipelines, and resolve performance bottlenecks with researchers and ML engineers.
- Improve training goodput by optimizing input processing systems for scalable, reliable distributed environments.
- Implement and maintain ML infrastructure tools; evaluate technologies and promote distributed-systems best practices.
What they ask for
- B.S. in Computer Science or Math, or 5+ years of equivalent real-world experience.
- Proficiency in distributed systems design and understanding of ML data pipeline optimization.
- Experience with TensorFlow and JAX; hands-on experience with Grain or tf.data service.
- Solid Python and C++ programming skills and practical familiarity with profiling tools.
Nice to have
- M.S. in Computer Science or Math.
- Familiarity with distributed dataflow frameworks such as ML Pathways.
Pay
$238,000–$302,000 USD base salary
Tools
ML Pathways, Grain, JAX, TensorFlow, tf.data service, Python, C++
AI focus
Improves ML training infrastructure and data pipelines for models used in autonomous driving.
Model-written summary · the original posting is authoritative