Staff Software Engineer, ML Systems

Who you are

You have extensive experience in software engineering, particularly in machine learning systems, and have a strong understanding of distributed systems. Your background includes optimizing training loops and streamlining data ingestion processes, ensuring efficiency and reliability in machine learning pipelines. You are skilled in writing high-performance code that integrates model logic with large-scale infrastructure, and you thrive in collaborative environments where you can act as a bridge between product teams and core infrastructure.

You possess a deep understanding of machine learning concepts and have experience working with various machine learning frameworks. Your ability to influence strategic infrastructure roadmaps demonstrates your leadership skills and your commitment to enhancing the capabilities of your team. You are adept at problem-solving and can eliminate friction in build and version control workflows, making you a valuable asset to any engineering team.

Desirable

Experience with cloud platforms and tools that support machine learning workflows is a plus. Familiarity with version control systems and CI/CD practices will help you excel in this role. You are also encouraged to bring innovative ideas to the table, contributing to the continuous improvement of processes and systems.

What you'll do

In this role, you will own multiple product teams' machine learning systems, focusing on efficiency and reliability. You will partner with core infrastructure teams to influence the roadmap for compute, storage, and scheduling, ensuring that the infrastructure meets the needs of machine learning applications. Your work will involve optimizing training loops on distributed cluster management systems and streamlining data ingestion processes to enhance the overall performance of machine learning models.

You will act as a 'force multiplier' for ML researchers, allowing them to focus on developing novel model architectures while you handle the ecosystem that supports them. This includes writing high-performance code that binds model logic to proprietary infrastructure, ensuring seamless integration and functionality. You will also be responsible for identifying and eliminating friction points in build and version control workflows, enhancing the productivity of your team.

Collaboration is key in this role, as you will work closely with various teams to ensure that machine learning systems are robust and scalable. You will participate in strategic discussions about infrastructure improvements and contribute to the overall vision of the company's machine learning initiatives. Your insights will help shape the future of autonomous driving technology at Waymo.

What we offer

Waymo offers a competitive salary range of $238,000—$302,000 USD, along with eligibility for an annual bonus program and equity incentive plan. You will also enjoy generous company benefits, subject to eligibility requirements. Joining Waymo means being part of a mission-driven company focused on improving mobility and safety through cutting-edge technology. You will have the opportunity to work with a talented team of engineers and researchers dedicated to making a significant impact in the field of autonomous driving.