Senior Software Engineer (Training Efficiency)

Join Waymo, a leader in autonomous driving technology. As a Senior Software Engineer, you will work on improving the runtime efficiency of input data pipelines for large-scale training workloads. You will collaborate with researchers and ML engineers, optimize input data processing systems, and implement advanced ML infrastructure tools. This is a unique opportunity to contribute to the development of cutting-edge ML systems and improve our model training processes. Enjoy a comprehensive benefits package, including medical, dental, and vision insurance, competitive compensation, and a hybrid work model. - Design and improve distributed input data pipelines for large-scale ML training workloads. - Collaborate with researchers and ML engineers to resolve bottlenecks in data pipeline performance. - Implement and maintain advanced ML infrastructure tools, including ML Pathways, Grain, JAX, and TensorFlow. - Proficient in distributed systems design with an understanding of ML data pipeline optimization - B.S. in Computer Science, Math, or 8+ years equivalent real-world experience - Practical familiarity with profiling tools to uncover performance bottlenecks - Experience with ML frameworks, including TensorFlow and JAX - Hands-on experience libraries like Grain or service - Solid programming skills in Python and C++ - MS in Computer Science, Math - Familiarity with distributed dataflow frameworks like ML Pathways