Build and operate petabyte-scale data systems and ML pipelines.
shepherd foundation models from research to robust components in Waymo Driver.
Create automated infrastructure for benchmarking, monitoring, and releasing models.
Work with large-scale compute and frameworks like Flume and JAX.
Improve speed, reliability, and efficiency of the end-to-end ML lifecycle.
Collaborate with AI Foundations, ML, and Platform teams to translate model innovations into on-road

🎯 Requirements

Masters degree in CS, ML, Robotics or equivalent experience.
Proficiency in Python.
Proficiency in C++.
Familiarity with modern DL frameworks (PyTorch, JAX, TensorFlow).
Experience building or maintaining large-scale data pipelines or ML infrastructure (e.g., Flume

🎁 Benefits

Discretionary annual bonus program and equity incentive plan.
Generous company benefits.

Apply on employer's website

This employer gathers applications via their own applicant tracking system.

You will be redirected to an external application form.

Share job

Meet JobCopilot: Your Personal AI Job Hunter

Automatically Apply to Engineering Jobs. Just set your preferences and Job Copilot will do the rest — finding, filtering, and applying while you focus on what matters.

Activate JobCopilot