Machine Learning Engineer

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 workflows. You are proficient in Python and have hands-on experience with frameworks like TensorFlow, which you have used to develop and deploy machine learning models. Your familiarity with cloud platforms, particularly AWS, allows you to efficiently manage resources and optimize costs associated with ML operations. You understand the importance of monitoring and reporting on resource usage, and you are skilled at developing dashboards that provide actionable insights to stakeholders. You thrive in collaborative environments and enjoy partnering with cross-functional teams to enhance developer velocity and streamline ML processes.

Desirable

Experience with Kubernetes for orchestration of ML workloads is a plus, as is familiarity with MLflow for managing the ML lifecycle. You have a keen interest in optimizing ML resource usage and are always looking for innovative ways to improve efficiency. Your analytical mindset enables you to identify trends and opportunities for improvement in ML operations, making you a valuable asset to any team.

What you'll do

In this role, you will analyze and report on ML resource usage, including compute, storage, and accelerators, across various teams and projects. You will develop, maintain, and improve dashboards and tools that monitor key resource metrics, providing insights that drive decision-making. Your work will involve close collaboration with the modeling team to implement solutions that enhance developer velocity and streamline complex ML workflows. You will be responsible for building automation and orchestration solutions that make ML processes more manageable and reliable. Additionally, you will investigate opportunities for optimizing resource usage and costs, ensuring that the ML operations at Waymo are efficient and effective.

What we offer

Waymo offers a competitive salary range of $170,000—$216,000 USD, along with eligibility for an annual bonus program and equity incentive plan. You will be part of a mission-driven company focused on improving access to mobility and saving lives through autonomous driving technology. The work environment is collaborative and innovative, with opportunities for professional growth and development. Waymo values diversity and encourages applicants from all backgrounds to apply, even if they do not meet every requirement listed.