The Agent Simulation group at Zoox is in search of machine learning engineers to play a crucial role in enhancing agent behaviors, assessing performance, and developing tools to prioritize simulation results.
You will be instrumental in the development of reliable and validated simulations, and have the opportunity to work with a wealth of real-world driving data and an exceptional infrastructure for testing and validating your algorithms.
Our ML group is focused on creating diverse and innovative agent behaviors, and your contributions will be key in this endeavor.
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, useable 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 and PyTorch, TensorFlow or JAX
- Strong 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.
Zoox also offers a comprehensive package of benefits including paid time off ( 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.
About Zoox Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market.
Sitting at the intersection of robotics, machine learning, and design, Zoox aims to provide the next generation of mobility-as-a-service in urban environments.
We’re looking for top talent that shares our passion and wants to be part of a fast-moving and highly execution-oriented team.
Accommodations If you need an accommodation to participate in the application or interview process please reach out to or your assigned recruiter.
AIML - Machine Learning Engineer/Scientist, Siri and Information Intelligence
As part of this group, you will be doing large scale machine learning and deep learning to improve Open Domain Question Answering and developing fundamental building blocks needed for Artificial Intelligence. We are looking for people with excellent applied machine learning experience and solid engi...
Machine Learning Engineer, Shop Ads - USDS
Apply state-of-the-art machine learning techniques to optimize advertisers' marketing strategies. The teams within USDS that deliver on this commitment daily span across Trust & Safety, Security & Privacy, Engineering, User & Product Ops, Corporate Functions and more. Familiarity with at...
Staff Machine Learning Engineer - Recommender Systems
We are looking for a passionate Machine Learning Engineer to build and scale the DTC personalization systems and services for our new global streaming app, Max, as well as any future DTC streaming apps. Good practical knowledge in modern machine learning lifecycle. You will be in the unique role of ...
Machine Learning Engineer - E-commerce Merchant and Creator Growth
Machine Learning Engineer, Data Scientist, and experience in causal machine learning is preferred. We are seeking a talented and motivated Machine Learning Engineer with expertise in marketplace growth to join our dynamic and fast-paced team. Apply feature engineering techniques to derive relevant f...
Machine Learning Engineer, Amazon
We are looking for a Machine Learning Engineer (MLE/SDE) to join the team to drive key science methods and delivery of the project, working closely with product and engineering leads, as well as our leadership. As a Machine Learning Engineer on this team, you will:. Lead end-to-end Machine Learning ...
Machine Learning Engineer, 3D Simulation
The 3D Simulation group at Zoox is looking for machine learning engineers to bring the latest research in 3D vision to improve diversity and blur the line between simulation and reality. Research, implement, and optimize state of the art machine learning approaches to improve simulation variation an...
Tech Lead, Senior Machine Learning Engineer - Ads Signal
As a Tech Lead, Machine Learning Engineer on the Ads Measurement team, you will lead the team to develop novel machine learning solutions, build scalable tech foundations and launch various products to maximize signal values for ads in a privacy-preserving way. Responsible for developing machine lea...
(General Hire) Machine Learning Engineer Graduate (TikTok Recommendation) - 2025 Start (MS)
Responsibilities - What You'II Do- Build industry-leading recommendation system, improving user experience, content ecosystem and platform security;- Deliver end-to-end machine learning solution to address critical product challenges;- Own the full stack machine learning system and optimize algorith...
Machine Learning Engineer - E-commerce Risk Control
We achieve our mission by a) developing state-of-art Machine Learning (ML) solutions to prevent customers, SPs, and Tiktok from being impacted by bad actors' actions and practices to gain unfair business advantages; b) empowering Tiktok teams, both internal to GNE and external, to reduce risk attemp...
AIML - Sr Machine Learning Engineer, Data and ML Innovation
As a Sr Machine Learning Engineer on the team, you will build algorithms and data applications that leverage the power of data (real and synthetic) to improve model training efficiency and performance. Are knowledgeable in classic machine learning algorithms (SVM, Random Forest, Naive Bayes, KNN etc...