Senior Software Engineer, Training Efficiency

Who you are

You have 5+ years of experience in software engineering, particularly in building and optimizing data pipelines for machine learning applications. Your background includes working with distributed systems, and you understand the complexities involved in large-scale ML training workloads. You are proficient in Python and have a solid grasp of machine learning concepts, which allows you to collaborate effectively with researchers and ML engineers. You are detail-oriented and have a knack for identifying bottlenecks in data pipeline performance, ensuring that systems run efficiently and effectively.

You thrive in collaborative environments and enjoy working closely with cross-functional teams. Your communication skills enable you to articulate technical challenges and solutions clearly, fostering a productive dialogue with both technical and non-technical stakeholders. You are passionate about improving processes and are always looking for innovative ways to enhance the efficiency of ML systems. You understand the importance of data quality and are committed to maintaining high standards in your work.

Desirable

Experience with cloud platforms such as AWS or GCP is a plus, as is familiarity with containerization technologies like Docker. Knowledge of additional programming languages or frameworks related to machine learning can further enhance your contributions to the team.

What you'll do

In this role, you will design and improve distributed input data pipelines that support large-scale ML training workloads. You will collaborate with researchers and ML engineers to identify and resolve performance bottlenecks, ensuring that the data pipelines operate at optimal efficiency. Your work will directly impact the performance of the Waymo Driver, contributing to the advancement of autonomous driving technology.

You will be responsible for analyzing existing data pipeline architectures and proposing enhancements that improve runtime goodput. This involves a deep understanding of the data flow and processing requirements for machine learning models. You will also engage in code reviews and provide mentorship to junior engineers, sharing your expertise and fostering a culture of continuous improvement within the team.

Your contributions will help shape the future of autonomous driving, as you work on systems that process vast amounts of data collected from real-world driving scenarios. You will have the opportunity to collaborate with some of the brightest minds in the field, driving innovation and pushing the boundaries of what is possible in autonomous technology.

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 benefit from a generous company benefits program, which includes health insurance, retirement plans, and other perks designed to support your well-being and professional growth. Join us at Waymo, where you can make a meaningful impact on the future of transportation and mobility.