Senior Software Engineer, ML Infrastructure, Predictive Planner

Waymo; California, Mountain View (MO, CA); On-site; USD 204,000 - 259,000; Full time

Job summary

A forward-thinking company is seeking a skilled ML Engineer to enhance their autonomous driving technology. In this exciting role, you will work with cutting-edge machine learning techniques, collaborating with talented data scientists and engineers to deploy robust ML models. Your expertise will drive the development of CI/CD pipelines and best practices for model governance, ensuring reliability in a fast-paced environment. Join a team that's redefining mobility and making a significant impact on safety and accessibility. If you are passionate about innovation and technology, this opportunity is perfect for you.

Benefits

Bonuses

Equity

Comprehensive Benefits

Qualifications

• 5+ years of experience in high-scale distributed or ML inference systems.
• Proficient in Python and C++ coding and testing.

Responsibilities

• Collaborate with teams to deploy ML models into production.
• Develop and maintain CI/CD pipelines for automation.

Skills

Python

C++

Machine Learning

CI/CD Pipelines

ML Ops

DevOps

Education

B.S. in Computer Science

Equivalent Experience

Tools

GCP

AWS

Kubernetes

Jenkins

TensorFlow

JAX

Flax

Beam

Kubeflow

Sagemaker

Vertex AI Pipelines

Job description

Waymo is an autonomous driving technology company with the mission to be the most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver - the world's most experienced driver - to improve access to mobility while saving lives lost to traffic crashes. The Waymo Driver powers Waymo One, a fully autonomous ride-hailing service, and can be applied to various vehicle platforms and use cases. It has completed over one million rider-only trips, autonomously driving tens of millions of miles on public roads and billions in simulation across 13+ U.S. states.

The Waymo Driver is powered by machine learning, and with recent advances, we aim to solve the next frontier of planning problems. Our work involves exciting aspects of ML, including Large Language Models (LLMs), Reinforcement Learning (RL), and vision understanding. In this role, you will ensure our systems remain reliable amidst rapid, experiment-driven development.

You will:

1. Collaborate with data scientists, engineers, and stakeholders to deploy ML models into production.
2. Develop and maintain CI/CD pipelines for automating data generation, model training, testing, and deployment.
3. Implement best practices for model versioning, monitoring, and governance.
4. Troubleshoot and resolve deployment and performance issues.
5. Stay current with emerging ML Ops and DevOps technologies and trends.
6. Apply best practices used in the Alphabet stack of ML technologies (e.g., TensorFlow, JAX, Flax, Beam).

You have:

1. B.S. in Computer Science, Math, or equivalent experience.
2. 5+ years of building and maintaining high-scale distributed or ML inference systems.
3. Proficiency in Python and C++ coding and testing.
4. Experience with large-scale fleet management, testing, and deployment tools like GCP, AWS, Kubernetes, Jenkins, GCS/S3.
5. Familiarity with MLOps tools such as tf.serving, Torchserve, Kubeflow, Sagemaker, Vertex AI Pipelines.
6. Understanding of ML fundamentals and experience with frameworks like JAX, PyTorch, or TensorFlow.

Preferred:

• Experience deploying ML models for computer vision, NLP, or recommendation systems.

The expected base salary range for this full-time position across US locations is $204,000 — $259,000 USD. Actual pay depends on factors like location, experience, education, and skills. Waymo offers bonuses, equity, and comprehensive benefits.