Demo

Principal Machine Learning Engineer - Prediction

Zoox
San Mateo, CA Full Time
POSTED ON 3/1/2025
AVAILABLE BEFORE 5/7/2025

The Prediction & Behavior ML team is responsible for developing machine learning (ML) algorithms that learn and predict behaviors from data, applying them both on-vehicle to influence driving behavior and off-vehicle to provide ML capabilities to simulation and validation. Given the tight integration of behavior forecasting and motion planning, our team collaborates closely with the Planner team to advance overall vehicle behavior. We also work closely with our Perception, Simulation, and Systems Engineering teams to accelerate our ability to validate our driving performance.

As a Principal ML Engineer, you will lead the development of machine learning algorithms that can range from influence from onboard autonomy to offboard autonomy and validation. You will collaborate closely with teams specializing in Perception, Simulation, and Safety Validation, influencing our overall technical stack. Your role will look at problems in a way that crosses team boundaries to prototype new approaches that influence the long term technical direction of multiple organizations within the company. The impact of the role can be in the form of impacting immediate company milestones to leading forward-looking exploratory projects.

Responsibilities

  • Develop new algorithms to model the future behavior of our own vehicle's future actions, both in predicting our driving trajectories and estimating their quality in relation to our goals of safety, progress, and comfort
  • Build the foundation models for the on-vehicle and offline applications
  • Develop new algorithms to apply generative deep learning to simulation to improve the realism of our offline validation systems
  • Leverage our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field
  • Provide technical mentorship to the broader group of ML developers at Zoox
  • Collaborate with engineers on Perception, Planning, and Simulation to solve the overall Autonomous Driving problem in complex urban environments

Qualifications

  • BS, MS, or PhD degree in computer science or related field
  • Experience with training and deploying Deep Learning models
  • Experience with production Machine Learning pipelines : dataset creation, training frameworks, metrics pipelines
  • Fluency in C or Fluency in Python with a basic understanding of C
  • Extensive experience with programming and algorithm design
  • Strong mathematics skills
  • 10 years of experience
  • Bonus Qualification

  • Conference or Journal publications in Machine Learning or Robotics related venues
  • Prior experience with Prediction and / or autonomous vehicles or robotics in general
  • Compensation

    There are three major components to compensation for this position : salary, Amazon Restricted Stock Units (RSUs), and Zoox Stock Appreciation Rights. The salary will range from $296,000-$451,000. A sign-on bonus may be part of a compensation package. Compensation will vary based on geographic location, job-related knowledge, skills, and experience.

    Zoox also offers a comprehensive package of benefits including paid time off (e.g. sick leave, vacation, bereavement), unpaid time off, Zoox Stock Appreciation Rights, Amazon RSUs, health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance.

    Salary : $296,000 - $451,000

    If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
    Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

    What is the career path for a Principal Machine Learning Engineer - Prediction?

    Sign up to receive alerts about other jobs on the Principal Machine Learning Engineer - Prediction career path by checking the boxes next to the positions that interest you.
    Income Estimation: 
    $149,493 - $192,976
    Income Estimation: 
    $184,796 - $233,226
    Income Estimation: 
    $149,493 - $192,976
    Income Estimation: 
    $184,796 - $233,226
    Income Estimation: 
    $77,900 - $95,589
    Income Estimation: 
    $101,387 - $124,118
    Income Estimation: 
    $101,387 - $124,118
    Income Estimation: 
    $119,030 - $151,900
    Income Estimation: 
    $119,030 - $151,900
    Income Estimation: 
    $149,493 - $192,976
    View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

    Job openings at Zoox

    Zoox
    Hired Organization Address Boston, MA Full Time
    The Detection and Optimization team in Perception is responsible for designing neural network model architectures that a...
    Zoox
    Hired Organization Address San Mateo, CA Intern
    As a Systems Integration Intern on the Software HIL (Hardware in the Loop) team, you will be a key part of the testing i...
    Zoox
    Hired Organization Address San Mateo, CA Full Time
    The Staff Failure Analysis Engineer will be responsible for working with the different teams to identify failure modes a...
    Zoox
    Hired Organization Address San Mateo, CA Full Time
    The Zoox Strength and Durability attribute team focuses on solving the unique challenge of delivering reliable service i...

    Not the job you're looking for? Here are some other Principal Machine Learning Engineer - Prediction jobs in the San Mateo, CA area that may be a better fit.

    Principal, Machine Learning Engineer

    Viant, Inc., San Francisco, CA

    AI Assistant is available now!

    Feel free to start your new journey!