ML Engineer, Foundation Model Infrastructure

Waymo is seeking an ML Engineer to build and operate large-scale data systems and ML pipelines for foundation model development in autonomous driving. You will be responsible for moving models from research to production, creating automated infrastructure for benchmarking and monitoring, and optimizing the ML development lifecycle using tools like Flume and JAX.

• Build and operate petabyte-scale data systems and ML pipelines
• Shepherd foundation models from research prototypes to production
• Create automated infrastructure for benchmarking and monitoring
• Utilize large-scale compute and frameworks like Flume and JAX
• Drive improvements in ML development lifecycle speed and reliability
• Partner with AI Foundations, ML, and Platform experts