Job Description
📋 Description 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.
