Machine Learning Engineer, ML Flywheel

Who you are

You have a strong background in machine learning and software engineering, with experience in building and maintaining data pipelines that support ML development. You are proficient in Python and have hands-on experience with ML frameworks such as TensorFlow. You understand the intricacies of deploying machine learning models and have worked with containerization technologies like Docker and orchestration tools such as Kubernetes. You are familiar with MLflow or similar tools for managing the ML lifecycle, including experimentation, reproducibility, and deployment. You thrive in collaborative environments and enjoy working closely with cross-functional teams to enhance developer velocity and streamline ML workflows. You are detail-oriented and have a passion for improving the reliability and scalability of ML systems.

Desirable

Experience with cloud platforms such as AWS or GCP is a plus, as is familiarity with data processing frameworks like Apache Spark. You have a keen interest in the latest advancements in machine learning and are eager to apply them in real-world applications. You are comfortable with version control systems like Git and have experience in CI/CD practices for ML projects.

What you'll do

In this role, you will implement new features in Waymo's data flywheel to scalably process ML data, ensuring that the data pipelines are efficient and reliable. You will develop robust ML workflows that continuously and automatically train, evaluate, and deploy models for both onboard and offboard applications. Your responsibilities will include supporting the reliable operation of ML pipelines at scale, troubleshooting issues, and optimizing performance. You will collaborate closely with the ML modeling team to understand their needs and provide the necessary infrastructure to enhance their productivity. You will also participate in code reviews and contribute to the overall improvement of the team's development practices. Your work will directly impact the efficiency of Waymo's autonomous driving technology, making it safer and more accessible.

What we offer

Waymo offers a competitive salary range of $170,000—$216,000 USD, along with participation in a discretionary annual bonus program and an equity incentive plan. You will have access to generous company benefits, including health and wellness programs, professional development opportunities, and a collaborative work environment that encourages innovation and growth. We believe in the importance of work-life balance and offer flexible working arrangements to support our employees' needs. Join us in our mission to improve access to mobility and save lives through cutting-edge technology.