Senior Software Engineer, Training Efficiency

Waymo

Mountain View, US
$238,000—$302,000 usd; bonus/equity: discretionary...
**
Distributed input data pipelines
Ml training workloads
Input data processing systems
** Waymo is seeking a Senior Software Engineer for their Training Efficiency team to enhance the runtime efficiency of input data pipelines for large-scale machine learning workloads. The ideal candidate will have experience in distributed systems design and machine learning frameworks, particularly TensorFlow and JAX. **

Job Summary

  • Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver.
  • You will improve the runtime efficiency of input data pipelines for large-scale training workloads.
  • This is a unique opportunity to work on ML systems and improve on our model training processes.

Matching Summary

Match Score: 75

** Waymo is seeking a Senior Software Engineer for their Training Efficiency team to enhance the runtime efficiency of input data pipelines for large-scale machine learning workloads. The ideal candidate will have experience in distributed systems design and machine learning frameworks, particularly TensorFlow and JAX. **

Salary

$238,000—$302,000 USD; Bonus/Equity: Discretionary annual bonus program, equity incentive plan; Benefits: Generous Company benefits program

Skills & Requirements

Must-have

  • Distributed input data pipelines
  • ML training workloads
  • Input data processing systems
  • ML infrastructure tools
  • Python and C++ programming
  • Profiling tools for performance

Nice-to-have

  • ML Pathways familiarity
  • Grain or tf.data service experience
  • Modern technology integration

Key Requirements

  • B.S. in Computer Science, Math, or 5+ years equivalent experience
  • Proficient in distributed systems design
  • Experience with ML frameworks (TensorFlow, JAX)
  • Hands-on experience with libraries like Grain or tf.data service
  • Solid programming skills in Python and C++
  • Practical familiarity with profiling tools

Work Rights

Not specified

Tailored Resume

Cover Letter