Machine Learning Engineer, Agent Simulation

Machine Learning Engineer, Agent Simulation

Zoox

Location: Foster City, CA

Salary: $141,000 to $234,000

Employment Type: Full Time

Role Highlights

1. Python
2. C++
3. Computer Science
4. Data
5. Infrastructure
6. Testing
7. Research
8. Cloud
9. PhD
10. Mathematics
11. LLMs
12. Robotics
13. Google Analytics

Tools, Libraries and Frameworks

1. JAX
2. PyTorch
3. TensorFlow
4. NumPy

Description

The Agent Simulation group at Zoox seeks machine learning engineers to enhance agent behaviors, assess performance, and develop simulation tools. The job focuses on developing reliable simulations using real-world driving data and a robust testing infrastructure. Essential skills include training and deploying deep learning models, creating machine learning pipelines, and fluency in Python, NumPy, and machine learning frameworks like PyTorch, TensorFlow, or JAX. Strong mathematics skills are required, and experience with reinforcement learning, imitation learning, and autonomous vehicles is beneficial.

Zoox is at the forefront of developing fully autonomous vehicle fleets from the ground up, aiming to transform urban mobility through robotics, machine learning, and innovative design. The company's mission centers around making personal transportation safer, cleaner, and more enjoyable, leveraging simulation technology to enhance driving software and validate safety measures. By embracing diverse perspectives, Zoox fosters innovation required for success in the competitive autonomous vehicle market.