companyWaymo LLC logo

Senior Software Engineer, Training Efficiency

Waymo LLCMountain View, California
On-site Full-time $238K/yr - $302K/yr

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Senior

Qualifications

Key Responsibilities:Design and optimize distributed input data pipelines for extensive ML training workloads. Work in tandem with researchers and ML engineers to identify and resolve performance bottlenecks in data pipelines. Enhance the runtime goodput of ML training workloads, optimizing input data processing systems while ensuring scalability and reliability in distributed environments. Implement and sustain advanced ML infrastructure tools, such as ML Pathways, Grain, JAX, and TensorFlow. Assess and incorporate modern technologies to improve ML system performance and scalability. Advocate for best practices in distributed systems architecture and provide technical leadership within the team. Qualifications:Bachelor's degree in Computer Science, Mathematics, or equivalent practical experience (5+ years). Expertise in distributed systems design with a focus on ML data pipeline optimization. Experience with ML frameworks, particularly TensorFlow and JAX. Hands-on experience with libraries like Grain or tf.data service. Strong programming skills in Python and C++. Familiarity with profiling tools to identify performance bottlenecks.

About the job

Waymo is a pioneering company in the realm of autonomous driving technology, striving to become the most reliable driver globally. Since its inception as the Google Self-Driving Car Project in 2009, Waymo has been dedicated to developing the Waymo Driver—The World's Most Experienced Driver™—which enhances mobility access while significantly reducing traffic-related fatalities. The Waymo Driver underpins our fully autonomous ride-hail service and can be utilized across various vehicle platforms and use cases. With over ten million rider-only trips completed, our technology has autonomously navigated over 100 million miles on public roads and tens of billions in simulated environments across more than 15 U. S. states.

Role Overview: As a key member of the Waymo ML Infrastructure team, you will collaborate with both Research and Production teams to refine models in Perception and Planning that are vital to our autonomous driving software. You will provide the optimal solutions for the entire model development lifecycle, developed in close partnership with Google teams. Your primary focus will be on enhancing the runtime efficiency of input data pipelines for large-scale training workloads, presenting a unique opportunity to advance ML systems and improve our model training processes.

About Waymo LLC

Waymo LLC is at the forefront of autonomous driving technology, with a mission to redefine transportation through innovation and safety. Our commitment to developing the most advanced self-driving systems is reflected in our extensive experience and proven track record in the industry.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.