Senior/Machine Learning Engineer, Agent Simulation
Listed on 2026-06-18
-
Software Development
Machine Learning/ ML Engineer, AI Engineer (Applied/Software)
Senior/Staff Machine Learning Engineer, Agent Simulation
Location:
San Francisco Bay Area
Compensation: $200, USD / year
AboutThe Role
The Agent Simulation group at Zoox is searching for machine learning engineers to play a crucial role in enhancing agent behaviors, assessing performance, and developing tools to prioritize simulation results. Your contributions will be key in creating diverse and innovative agent behaviors using real-world driving data and our infrastructure for testing and validating algorithms.
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
- 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 and PyTorch, Tensor Flow or JAX
- Strong mathematics skills
- 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
Salary ranges from $200,000 to $300,000. Compensation also includes Amazon Restricted Stock Units (RSUs) and Zoox Stock Appreciation Rights. Signing bonuses may be part of the package. Compensation varies by location, experience, and related job knowledge.
Benefits include paid time off (sick leave, vacation, bereavement), unpaid time off, health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance.
AccommodationsIf you need an accommodation to participate in the application or interview process, please reach out to or your assigned recruiter.
#J-18808-Ljbffr(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).