Waymo · 3 weeks ago
Machine Learning Engineer, Model Optimization
Mountain View, CA
Full-time
Onsite
Mid Level
$170K/yr - $216K/yr
3+ years expWaymo is an autonomous driving technology company with the mission to be the world's most trusted driver. The role involves optimizing model training and inference for the Waymo Driver, analyzing real-time inference of complex model architectures, and collaborating with various teams to enhance the performance of autonomous driving systems.
Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
Responsibilities
Optimize FLOPs utilization in model training and model inference through model architecture/ hardware co-development, optimize for a naturally sparse representation (most spatial-temporal information in self-driving is sparse)
Optimize model inference for different onboard and offboard (simulation) platforms
Analyze and optimize real-time inference of complex model architectures with many model components as well as on the critical path within an onboard system
Qualification
Required
Bachelors in Computer Science or a similar discipline, or an equivalent amount of deep learning experience
3+ years experience in Machine Learning and/or Computer Vision
Experience with Python
Experience with ML frameworks like PyTorch or JAX
Preferred
MS or PhD Degree in Machine Learning, Robotics, Computer Science or a similar discipline
Publications at top-tier conferences like CVPR, ICCV, ECCV, ICLR, ICML, ICRA, IROS, RSS, NeurIPS, AAAI, IJCV, PAMI
Experience with C++
Benefits
Waymo’s discretionary annual bonus program
Equity incentive plan
Generous Company benefits program
Company
Waymo
Waymo is a mobility technology company that improves transportation by developing self-driving solutions for travelers and daily commuters. It is a sub-organization of Alphabet.
1001-5000 employees
H1B Sponsorship
Waymo has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (231)
2024 (175)
2023 (268)
2022 (306)
2021 (298)
2020 (317)
Funding
Current Stage
Late StageTotal Funding
$11.1BKey Investors
Alphabet
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
2020-05-12Series A· $750M
Leadership Team
Tekedra Mawakana
Co-Chief Executive Officer
Elisa de Martel
Chief Financial Officer
Recent News
2026-01-12
PYMNTS.com
2026-01-12
Company data provided by crunchbase