Staff Machine Learning Engineer, Runtime & Optimization
Description For Staff Machine Learning Engineer, Runtime & Optimization
Waymo, formerly the Google Self-Driving Car Project, is at the forefront of autonomous driving technology since 2009. As a Staff Machine Learning Engineer on the ML Platform team, you'll play a crucial role in optimizing and improving the compute performance of ML systems both in the cloud and on self-driving vehicles.
The position offers an exciting opportunity to work with world-class ML scientists across perception, planning, research, and simulation teams. You'll be responsible for leading technical initiatives that enhance ML workload efficiency, applying sophisticated optimization techniques, and managing projects from conception to deployment.
The role combines deep technical expertise in machine learning systems with leadership responsibilities. You'll be working with cutting-edge ML frameworks like JAX, XLA, Triton, and CUDA, while also mentoring junior engineers and driving technical decisions that impact Waymo's autonomous driving capabilities.
Your work will directly contribute to making Waymo's ML systems more efficient and effective, ultimately advancing the company's mission of building the World's Most Experienced Driver™. The position offers competitive compensation, comprehensive benefits, and the chance to work on technology that's transforming transportation.
The ideal candidate brings both technical depth in ML systems and leadership experience, with the ability to balance complex technical challenges with team development and project management. You'll be part of a team that's not just developing ML solutions, but fundamentally changing how autonomous vehicles perceive and navigate the world.
Responsibilities For Staff Machine Learning Engineer, Runtime & Optimization
- Lead collaboration with Waymo ML scientists in perception, planner, research and simulation
- Identify opportunities in systems and models to make ML workloads faster
- Lead projects from proposals through execution by developing junior engineers
- Analyze and improve ML system workloads on both cloud and self-driving cars
- Apply model optimization, efficient deep learning techniques and ML software improvements
Requirements For Staff Machine Learning Engineer, Runtime & Optimization
- M.S. in CS, EE, Deep Learning or a related field
- 2+ years of experience as a technical lead
- 5+ years of experience developing solutions in ML systems or ML software stack
- Deep understanding of ML system architecture, performance analysis and tools
- Strong Python or C++ programming skills
- Experience writing project plans, engaging with customer teams, mentoring
- Responsible for goals & execution, reporting status
Benefits For Staff Machine Learning Engineer, Runtime & Optimization
- Top-notch medical, dental and vision insurance
- Mental wellness support
- Gym membership
- Special wellness programs
- Competitive compensation
- Bonus opportunities
- Equity
- Employee provident fund
- Employee discounts
- Flexibility to work from another location for four weeks per year
- Paid time off
- Bereavement leave
- Sick leave
- Parental leave
Jobs Related To Waymo Staff Machine Learning Engineer, Runtime & Optimization
