Senior Software Engineer (ML Evaluation Infra and Efficiency)

or a related field is preferred.

Join Waymo, the leading autonomous driving technology company, as a Senior Software Engineer. You will work closely with research and production teams to develop and deploy models for perception and planning. Your expertise in ML systems will help improve pre-trained models for deployment in Waymo Driver and future products. You will design and build distributed evaluation platforms, optimize performance, and collaborate with ML engineers to enhance the evaluation infrastructure. Enjoy a comprehensive benefits package, including medical, dental, and vision insurance, competitive compensation, and a hybrid work model.

• Design and build distributed evaluation platforms for large-scale ML evaluation workloads.
• Profile evaluation platforms, identify performance bottlenecks, and implement optimizations to improve inference speed and resource utilization.
• Collaborate with ML engineers to understand evaluation requirements and scenarios, and improve DevX of the evaluation infrastructure.
• Proficient in distributed systems design with an understanding of ML efficiency
• Experience with ML frameworks, including TensorFlow, JAX, XLA
• Solid programming skills in Python and C++
• Practical familiarity with profiling tools to uncover performance bottlenecks
• B.S. in Computer Science, Math, or 3+ years equivalent real-world experience
• Familiarity with large-scale evaluation platforms for LLMs
• MS in Computer Science, Math