Let’s get started
By clicking ‘Next’, I agree to the Terms of Service
and Privacy Policy, and consent to receive emails from Rise
Jobs / Job page
Machine Learning Engineer - Prediction and Planning image - Rise Careers
Job details

Machine Learning Engineer - Prediction and Planning

The Offline Driving Intelligence team is responsible for developing Foundation Models for prediction and planning, applying them both off-vehicle to provide ML capabilities to simulation and validation and on-vehicle to influence driving models. 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 Prediction and Planning Machine Learning Engineer you will work on the bleeding edge of the industry, developing novel machine learning pipelines and models to predict the behavior of other agents in the world and planning the best course of action for the ego vehicle.


In this role, you will:
  • Develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for our autonomous vehicle. You will also work on novel techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort, compliance.
  • Contribute to our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field
  • Develop metrics and tools to analyze errors and understand improvements of our systems
  • Collaborate with engineers on Perception, Planning, and Simulation to solve the overall Autonomous Driving problem in complex urban environments


Qualifications:
  • MS, or PhD degree in computer science or related field with 5 or more years of industry experience
  • Experience with training and deploying transformer-based model architectures and reinforcement learning techniques
  • 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


Bonus Qualifications:
  • Strong mathematics skills
  • Prior experience with Prediction and/or Planning for autonomous vehicles or robotics
  • Conference or Journal publications in Machine Learning or Robotics related venues


$204,000 - $245,000 a year
Base Salary Range

There are three major components to compensation for this position: salary, Amazon Restricted Stock Units (RSUs), and Zoox Stock Appreciation Rights. A sign-on bonus may be offered as part of the compensation package. The listed range applies only to the base salary. Compensation will vary based on geographic location and level. Leveling, as well as positioning within a level, is determined by a range of factors, including, but not limited to, a candidate's relevant years of experience, domain knowledge, and interview performance. The salary range listed in this posting is representative of the range of levels Zoox is considering for this position.

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.

Average salary estimate

$224500 / YEARLY (est.)
min
max
$204000K
$245000K

If an employer mentions a salary or salary range on their job, we display it as an "Employer Estimate". If a job has no salary data, Rise displays an estimate if available.

Similar Jobs
Photo of the Rise User

Lead Zoox's Robot Link Platform team managing backend services that enable autonomous vehicle communication and fleet operations.

Photo of the Rise User
Posted 13 days ago

Drive the electrical system design for Zoox's autonomous robotaxi as a Senior Electrical System Architect, integrating cutting-edge automotive components and ensuring system reliability.

Photo of the Rise User
Posted 10 days ago

An established engineering firm in Phoenix is looking for a skilled Transmission Line Engineer experienced in overhead line systems to lead and deliver critical power delivery projects.

Lead Yelp's globally distributed compute platform teams to drive innovation, reliability, and developer experience in a remote, collaborative environment.

Photo of the Rise User
Posted 13 days ago

A Timing Engineer role at Tenstorrent requires expertise in static timing analysis to optimize high-performance RISC-V chip designs in a collaborative hybrid environment.

Toyota Hybrid Saline, Michigan
Posted 7 days ago

Experienced Senior Engineer needed at Toyota to drive cost planning strategies and collaboration in the Cost Innovation Group for enhanced vehicle profitability.

Photo of the Rise User
VT Leeboy Hybrid Maiden, North Carolina, United States
Posted yesterday

A Manufacturing Engineer role at LeeBoy focusing on process improvement, product launch coordination, and efficiency optimization in a heavy equipment manufacturing environment.

Photo of the Rise User

Drive technical integration and client support as a Solutions Engineer for Sonio’s AI-powered prenatal diagnostic tools, working remotely across the West Coast.

KC Fixture & Display, Inc. Hybrid Kansas City, Missouri, United States
Posted 4 days ago

Lead the design process as a Design Engineer, transforming customer visions into innovative and manufacturable custom retail products at a creative and growth-focused company in Kansas City.

Posted 8 days ago

Lead critical reliability engineering programs at Johnson Matthey's Savannah facility to enhance equipment performance and operational excellence.

Photo of the Rise User
Posted 13 days ago

Contribute to cutting-edge robotics and embedded firmware development at Carbon Robotics to help build the future of sustainable farming technology.

Photo of the Rise User

Sargent & Lundy is seeking a Senior Electrical Engineer to deliver innovative electrical engineering solutions for critical government projects in a hybrid work environment.

Photo of the Rise User
AECOM Hybrid Baltimore, MD, United States
Posted 14 days ago

Lead engineering teams and manage high voltage utility transmission design projects at AECOM, a global infrastructure leader driving sustainable and innovative solutions.

Photo of the Rise User
Posted 4 days ago

Ava Labs is seeking an experienced Senior Platform Engineer to design and maintain cutting-edge blockchain platforms and CI/CD systems remotely from New York.

Photo of the Rise User

As a Senior Database Reliability Engineer at PointClickCare, you will manage and enhance our cloud-based database systems while driving automation and performance improvements within a pioneering health tech environment.

Zoox was founded to make personal transportation safer, cleaner, and more enjoyable—for everyone. To achieve that goal, the team created a whole new form of transportation. Zoox will provide mobility-as-a-service in dense urban environments.

28 jobs
MATCH
Calculating your matching score...
FUNDING
DEPARTMENTS
SENIORITY LEVEL REQUIREMENT
TEAM SIZE
EMPLOYMENT TYPE
Full-time, unknown
DATE POSTED
July 9, 2025
Risa star 🔮 Hi, I'm Risa! Your AI
Career Copilot
Want to see a list of jobs tailored to
you, just ask me below!