Machine Learning Engineer, Agent Simulation
Location: Seattle
Compensation: $134,000 - $222,000 USD / year
About The Role: Develop innovative agent behaviors to enhance autonomous mobility solutions in urban transportation.
In this role, you will:
  • Research, implement, and optimize state-of-the-art machine learning approaches to improve plausible agent behaviors
  • Find innovative solutions for agent behaviors and simulation analysis
  • Prove machine‑learned algorithms have better performance than heuristics
  • Leverage our large‑scale machine‑learning infrastructure to discover new solutions
  • Analyze the difference between behaviors in simulation and real-world
  • Work cross‑functionally with our safety and autonomy engineers
  • Build scalable, usable cloud pipelines for machine learning solutions for simulation
Qualifications:
  • BS, MS, or PhD degree in Computer Science or a related field
  • Experience with training and deploying deep learning models
  • Experience with production machine‑learning pipelines: dataset creation, training frameworks, metrics pipelines
  • Fluency in Python and a basic understanding of C++Fluency with NumPy, PyTorch, TensorFlow or JAXStrong mathematics skills
  • Bonus Qualifications
  • Experience with LLMs, reinforcement learning, imitation learning
  • Conference or journal publications in machine learning or robotics
  • Prior experience with agent behaviors, prediction, autonomous vehicles or robotics
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 $134,000 to $222,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. Benefits include paid time off (sick leave, vacation, bereavement), unpaid time off, Zoox Stock Appreciation Rights, Amazon RSUs, health insurance, long‑term care insurance, disability insurance, and life insurance. Accommodations If you need an accommodation to participate in the application or interview process please reach out to accommodations@zoox.com or your assigned recruiter.

Also on the board Same function, level within a rung

Level

Senior

Salary

$134,000 per year

Location

Seattle, WA

Occupation

Computer and Information Research Scientists

Industry

Research and Development in the Physical, Engineering, and Life Sciences (except Nanotechnology and Biotechnology)

Posted

yesterday

Apply for this role →