Machine Learning Engineer, Prediction & Planning jobs in United States
cer-icon
Apply on Employer Site
company-logo

Waymo · 3 months ago

Machine Learning Engineer, Prediction & Planning

Waymo is an autonomous driving technology company with a mission to be the world's most trusted driver. The role involves developing machine learning solutions for predicting future states and planning the behavior of autonomous vehicles, collaborating with researchers and engineers to enhance the capabilities of the Waymo Driver.

Artificial Intelligence (AI)AutomotiveAutonomous VehiclesSensorTransportation
check
H1B Sponsor Likelynote

Responsibilities

Develop the next-generation ML-powered prediction and planning system to enhance the performance and capabilities of the ML driver and support the rapid scaling of Waymo’s business
Frame open-ended, real-world challenges as well-defined ML problems; research, develop, and apply cutting-edge ML techniques, including foundation models and reinforcement learning, for the planning and prediction tasks of autonomous vehicles
Collaborate with world-class researchers, engineers and product owners to create safe, smooth planning behaviors for all road users and to meet product requirements
Develop and evaluate large models, and integrate them into Waymo’s production planning software for real-world applications through close partnership with the Planner and Research teams

Qualification

Machine LearningDeep LearningPythonReinforcement LearningTensorFlowJAXPytorchC/C++RoboticsAnalytical SkillsDebugging SkillsSoftware DevelopmentPublicationsPerformance Optimization

Required

BS in Computer Science, ML, Robotics, similar technical field of study, or equivalent practical experience
Demonstrated contributions to the ML community through publications, open-source projects, or significant industry impact
Hands-on experience with modern deep learning libraries (eg: TensorFlow, JAX, Pytorch)
Proficient programming skills (eg: Python, C/C++)
Strong analytical and debugging skills
5+ years of experience in Machine Learning modeling and/or Autonomous Vehicles

Preferred

MS or PhD in Computer Science, Machine Learning, Robotics, or a related field
Publications in top-tier conferences such as ICML, NeurIPS, CVPR, ICCV, ECCV, ICLR, IROS, CoRL, ACL, or EMNLP
Prior software development or ML research industry experience (including internships)
General software engineering experience solving motion planning or related robotics problems
Experience applying or evaluating ML-based systems in production environments
Experience with performance optimization of deep models, including with respect to specific hardware architectures

Benefits

Incredible maternity leave
Health care
Retirement planning
Waymo’s discretionary annual bonus program
Equity incentive plan
Generous Company benefits program

Company

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.

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 Stage
Total Funding
$11.1B
Key Investors
Alphabet
2024-07-23Series C· $5.6B
2021-06-16Series B· $2.5B
2020-05-12Series A· $750M

Leadership Team

leader-logo
Tekedra Mawakana
Co-Chief Executive Officer
linkedin
leader-logo
Annabel Chang
Head of State Policy & Government Relations
linkedin
Company data provided by crunchbase