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 $190,000 to $285,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.
Senior Staff Engineer - Microservices
A company is looking for a Senior Staff Engineer specializing in Software Development and Microservices. ...
Machine Learning Engineer Graduate (E-Commerce Supply Chain & Logistics - CV/Multimodal)- 2025 Start (PhD)
Join our vibrant and fast-paced team at TikTok as a Machine Learning Engineer, specializing in Computer Vision (CV), Artificial Intelligence Generated Content (AIGC), or Multimodal machine learning. We are seeking brilliant and motivated graduate software engineers, who are eager to apply their know...
Sr. Machine Learning Engineer
We're looking for a highly technical and mission-driven Senior Machine Learning Engineer to build real-time, low-latency machine learning services. The Machine Learning Engineering team builds Viant's proprietary machine learning system that will disrupt the Ad Tech industry. This individual will le...
Machine Learning Engineer - TikTok Search Algorithm (NLP, Relevance, Understanding, User Engagement)
On the TikTok Search Team, you will have the opportunity to develop and apply cutting edge machine learning technologies in real-time large-scale systems, which serve billions of search requests every day. Experience in one or more of the following areas is preferred: NLP, Ranking, Ads, search engin...
Machine Learning Engineer (Remote Eligible)
Machine Learning Engineer (Remote Eligible). Use pre-trained Generative AI models, deep learning models, or traditional machine learning models to solve customer problems. Partner effectively with software engineers and product managers to integrate the machine learning solution into the product. St...
Machine Learning Engineer (CV/NLP/Multimodal/LLM) -E-commerce Governance
Solid foundation in data structures/algorithms, proficient in machine learning/deep learning theory, and rich practical experience. Explore reinforcement learning and operational research algorithms in intelligent dispatching and intelligent audit scheduling to improve audit quality and efficiency. ...
Machine Learning Data Engineer Specialist
We are looking for a Senior Machine Learning Engineer to help us build a brand-new financial technology platform for the future. Develop, deploy, and scale machine learning models and applications in production and lower environments. Collaborate with data scientists to develop and test machine lear...
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...
(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, Search Ads - USDS
Good knowledge in one of the following areas: machine learning, deep learning, backend, large-scale systems, data science, full-stack. Responsibilities: Participate in the development of a large-scale Ads system• Responsible for relevance model and strategy optimization, such as semantic matching mo...