Senior Software Engineer, Waymo
ML Performance
$192-243k
+Equity incentive plan
Office located in Mountain View, CA
Autonomous driving technology
Job no longer available
Autonomous driving technology
1001+ employees
Job no longer available
$192-243k
+Equity incentive plan
Office located in Mountain View, CA
1001+ employees
Company mission
To make it simpler and safer for people to move things and themselves around.
Role
Who you are
- Advanced degree in Computer Science, Machine Learning, Robotics, similar technical field of study, or equivalent practical experience
- Experience with popular ML frameworks such as JAX, PyTorch, or TensorFlow
- Code and testing skills in Python
- Experience with GPU programming / optimization using CUDA or similar technologies
- Prior experience optimizing ML model performance e.g. quantization, architectural optimizations (e.g. kv-caching, flash attention, distillation) or low-level optimization (e.g. CUDA kernels, XLA)
Desirable
- PhD degree in Computer Science or a similar discipline, or an equivalent amount of deep learning experience
- 4+ years of computer science or deep learning experience
- Experience implementing benchmark models and improving their performance
- Experience with model optimization work deployed in industrial settings
- Low-level Tensorflow experience related to performance (e.g. memory management, real-time / batch processing use-cases)
- Code and testing skills in C++
What the job involves
- Report into the TLM of ML performance
- Profile and debug model performance bottlenecks
- Improve model and data ingestion operations to reduce compute costs
- Implement novel custom kernels to improve performance and cost
- Suggest alternative modeling architectures that balance performance and quality
- Design and implement benchmarks to track performance optimization
- Work across Planner, Research, Evaluation to improve both simulation and on-board performance
Our take
Waymo, Alphabet's self-driving car company, is changing the way people get around, Its robotaxis, which don't need a human driver, are already operating in cities like Phoenix, San Francisco, Los Angeles, Atlanta, and Austin. Soon, Waymo will expand to Detroit, Las Vegas, and San Diego, showing just how autonomous cars are moving from experiments to real-world transportation.
The company has already completed over 10M fully driverless rides and driven more than 100M miles. Its cars use advanced sensors and AI to navigate streets safely, while partnerships with Uber and DoorDash show that Waymo's technology works for both passengers and deliveries. And it's feeling ambitious too, hoping to offer one million trips per week by the end of 2026.
Even with competitors like Tesla and Zoox, Waymo is leading the autonomous ride-hailing market. Its cars have fewer crashes and keep passengers safer than traditional vehicles. With international plans for London and Japan, Waymo isn't just growing in the US, its setting its sights on shaping the future of transportation worldwide.
Steph
Company Specialist at Welcome to the Jungle
Insights
-1% employee growth in 12 months
Company
Funding (last 2 of 5 rounds)
Feb 2026
$16bn
LATE VC
Aug 2024
$5.6bn
SERIES C
Total funding: $27.1bn
Company benefits
- Medical, dental, and vision insurance for employees and dependents
- Employee assistance programs focused on mental health
- Personalized workplace adjustments for diverse needs and abilities, including physical, mental, and neurodivergent considerations
- Access to mental health apps
- Onsite wellness centers
- Second medical opinion for you and your loved ones
- Medical advocacy program for transgender employees
- Counseling services
- Support programs including menopause benefit
- Competitive compensation
- Regular bonus and equity performance grant opportunities
- Generous 401(k) and regional retirement plans
- Annual cross-company compensation review and pay equity analysis
- 1-on-1 financial coaching
- Fertility and growing family assistance
- Parental leave and baby bonding leave
- Caregiver leave
- Elder care and support
- Backup childcare
- Survivor income benefit
- Paid time off, including vacation, bereavement, sick leave, parental leave, disability, and holidays
- Jury duty leave
- Military leave
- Hybrid work model with remote work opportunities also available
- Educational reimbursement
- Peer learning and coaching platform
- Donation matching programs
- Employee resource groups
- Volunteer hours
- Internal community groups and local culture clubs
- Inspiring spaces to work, recharge, and collaborate with fellow Waymonauts
- On-site meals and snacks
- Fitness centers, massage programs, and ergonomic support
- On-demand fitness, wellbeing, and cooking classes
- Commuter benefits
Company HQ
Mountain View, CA
Leadership
Dmitri Dolgov
(Co-CEO)Dmitri joined Waymo in 2016 as CTO, becoming Co-CEO in 2021. Prior to this, he spent nearly 8 years at Google as a Distinguished Software Engineer and 2 years at Toyota Research Institute as a Senior Research Scientist.
Tekedra N. Mawakana
(Co-CEO)Tekedra joined Waymo in 2017 as the Global Head of Policy, holding various roles before becoming Co-CEO in 2021. Previously, she was the VP of Global Government Relations and Public Policy at eBay and VP and Deputy General Counsel of GLobal Public Policy at Yahoo.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 247 more jobs at Waymo