Machine Learning Data Infrastructure Engineer

About the role

About The Company

Waymo is a pioneering autonomous driving technology company committed to transforming mobility and saving lives through innovation. Originating from the Google Self-Driving Car Project in 2009, Waymo has established itself as a leader in developing the Waymo Driver, the world's most experienced autonomous driving system. With a mission to be the world's most trusted driver, Waymo continually advances its technology to improve access to safe and reliable transportation. The Waymo Driver powers Waymo's fully autonomous ride-hail service and is adaptable to a variety of vehicle platforms and use cases. Having completed over ten million rider-only trips and accumulated experience driving over 100 million miles on public roads across more than 15 states, Waymo leverages extensive real-world data and simulation to enhance its systems and ensure safety and efficiency in autonomous driving.

About The Role

We are seeking a talented and motivated Machine Learning Data Infrastructure Engineer to join our Perception team. In this hybrid role, you will work closely with cross-functional teams to develop scalable data collection, labeling, and evaluation systems that support the continuous improvement of the Waymo Driver. Your primary focus will be on creating large-scale datasets, developing automation pipelines for data mining and model updates, and designing infrastructure solutions that facilitate efficient model training and validation. You will play a vital role in optimizing the performance of perception models, ensuring they can handle the complexities of real-world environments. Reporting to a Technical Lead Manager, you will collaborate with ML infrastructure teams and problem domain specialists to streamline processes, address bottlenecks, and support the scaling of data needs for autonomous vehicle perception systems.

Qualifications

A degree in Computer Science, Engineering, or a related technical field

4+ years of professional experience in software engineering and machine learning

Proficiency in C++ and Python programming languages

Experience designing distributed systems for processing data at scale, particularly in ML data infrastructure

Strong foundational understanding of machine learning principles and state-of-the-art methods

Passion for building world-class ML infrastructure and systems

Excellent communication and collaboration skills

Responsibilities

Create large-scale datasets and develop training recipes for human and machine labeling processes

Design and implement automated data collection pipelines and model update flywheels for continuous learning

Develop evaluation methods to assess real-world model performance and detect regressions in updates

Understand data requirements of the problem domain team and design scalable infrastructure solutions to support model improvements and product expansion

Build and validate ML data infrastructure, ensuring it supports the scaling of VLM data needs

Collaborate with ML infrastructure teams and problem domain teams to resolve issues and optimize validation workflows

Benefits

Competitive salary range of $250,000 to $334,530 USD, commensurate with experience and location

Participation in Waymo’s discretionary annual bonus program and equity incentive plan

Comprehensive benefits package including health, dental, and vision insurance

Generous paid time off and flexible work arrangements

Opportunities for professional growth and development within a leading autonomous driving company

Equal Opportunity

Waymo is an equal opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all employees. All qualified applicants will receive consideration for employment without regard to race, color, religion, gender, gender identity or expression, sexual orientation, national origin, genetics, disability, age, or veteran status.