Senior / Staff Machine Learning Engineer - Behavior Models for Road Users
Zoox · Foster City, CA
Extracted from the posting
Develops learned behavior models for road users such as cars, bicycles, and pedestrians for Zoox simulation, using imitation learning, reinforcement learning, and generative models conditionable on scenario intent. Builds driving plan evaluation metrics and collaborates with Planner, Simulation, and Validation teams. Requires a PhD with 4+ years or a master's with 7+ years of experience.
This role in the Offline Driving Intelligence team is responsible for developing/learning behavior models for road users such as cars, bicycles, and pedestrians. These agents populate Zoox's simulations and must be indistinguishable from real road users, yet fully controllable: promptable into the rare, adversarial, safety-critical behaviors we need to test against. This means the team’s models directly impact how fast Zoox can train, validate and ship its driving stack. Our team collaborates closely with Planner, Simulation and Validation teams to develop and validate our driving performance. As an ML Agents 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...
Qualifications
Bonus Qualifications
- Top tier publications (NeurIPS, ICML, CVPR)
- Experience with JAX
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.
Compensation: USD 254,000 – 366,000 per year salary
Related jobs
Sr. Model Engineer, Voice
SpaceXAISr. Model Engineer, Imagine
SpaceXAI