Foundation ML Engineer: Scale & Deploy Large Models

Neura Market; Mountain View (CA); Hybrid; USD 175,000 - 215,000; Full time

Job summary

Waymo is seeking a scientist to advance the Waymo AI Foundations by building and operating data systems and ML pipelines that power foundation models for the Waymo Driver. You will contribute to transforming research prototypes into robust components used in autonomous driving applications.

The role reports to a Senior Research Scientist in a hybrid capacity, collaborating with multiple ML and platform teams to push the boundaries of reinforcement learning, learning from demonstration, and

Qualifications

• Masters degree in Computer Science, Machine Learning, Robotics or equivalent practical experience.
• Proficiency in Python.
• Proficiency in C++.
• Familiarity with one of the modern deep learning frameworks (e.g. PyTorch, JAX, TensorFlow).
• Experience building or maintaining large-scale data pipelines or ML infrastructure (e.g. Flume, Spark, Borg, Kubeflow).

Responsibilities

• Build and operate the petabyte-scale data systems and ML pipelines at the heart of Waymo's foundation model development.
• Shepherd cutting-edge foundation models from research prototypes to robust components within the Waymo Driver.
• Create the automated infrastructure for rigorously benchmarking, continuously monitoring, and safely releasing models.
• Wield large-scale compute and frameworks like Flume and JAX to process massive datasets and train/deploy complex models.
• Drive significant leaps in the speed, reliability, and efficiency of the end-to-end ML development lifecycle.
• Partner with AI Foundations, ML, and Platform experts to transform model innovations into tangible on-road improvements.

Skills

Python

C++

ML frameworks

Education

Masters degree in CS/ML/Robotics or equivalent

Tools

PyTorch

JAX

TensorFlow

Flume

Spark

Borg

Kubeflow

Job description

Waymo is seeking a scientist to advance the Waymo AI Foundations by building and operating data systems and ML pipelines that power foundation models for the Waymo Driver. You will contribute to transforming research prototypes into robust components used in autonomous driving applications.

The role reports to a Senior Research Scientist in a hybrid capacity, collaborating with multiple ML and platform teams to push the boundaries of reinforcement learning, learning from demonstration, and