ML Engineer (AI)

TL;DR

ML Engineer (AI): Building and operating petabyte-scale data systems and ML pipelines for foundation model development with an accent on research-to-production transition and automated infrastructure. Focus on optimizing the ML development lifecycle and scaling complex models using JAX and Flume.

Location: On-site/Hybrid in Mountain View, California; must be based in the US

Salary: $175,000—$215,000 USD

Company

Autonomous driving technology company focused on building the world's most experienced and trusted driver.

What you will do

• Build and operate petabyte-scale data systems and ML pipelines for foundation model development.
• Transition cutting-edge foundation models from research prototypes to robust production components.
• Develop automated infrastructure for benchmarking, continuous monitoring, and safe model release.
• Process massive datasets and train complex models using large-scale compute and frameworks like Flume and JAX.
• Drive improvements in the speed, reliability, and efficiency of the end-to-end ML development lifecycle.
• Collaborate with AI Foundations and Platform experts to turn model innovations into on-road improvements.

Requirements

• Master's degree in Computer Science, Machine Learning, Robotics, or equivalent practical experience.
• Proficiency in Python and C++.
• Experience with modern deep learning frameworks such as PyTorch, JAX, or TensorFlow.
• Experience building or maintaining large-scale data pipelines or ML infrastructure (e.g., Spark, Kubeflow).
• Must be based in the US

Nice to have

• Strong software engineering skills for developing large, complex shared codebases.
• Experience in autonomous vehicle planning and related research.
• Experience designing distributed systems or MLOps platforms (model versioning, CI/CD for ML).
• Prior experience in industrial or research settings developing ML model evaluation methodologies.

Culture & Benefits

• Participation in a discretionary annual bonus program.
• Equity incentive plan.
• Generous company benefits program.
• Opportunity to work on world-leading autonomous driving technology within the Alphabet ecosystem.