Senior Machine Learning Engineer - Perception Detection and Tracking

Job Description

Job Description

The Object Detection and Tracking team at Zoox deals with the perception of all people and objects that can move. Your role is to work on the ML model team to develop detection and tracking models and to integrate them into the vehicle stack.

In this role, you will have access to the world's best sensor data and an incredible infrastructure for testing and validating your algorithms. We are creating new algorithms for detection, tracking, classification, and high-level scene understanding, and you could work on any (or all!) of these components.

In this role, you will...

• Define on-vehicle architecture for producing the core tracking results from the Perception stack.
• Assist with the implementation of core tracking results from the Perception stack 
• Work with both the model and data teams to develop a high-performance, efficient model that can run on a vehicle.
• Work with Perception data, both on the input and output of machine learning models

Qualifications

• BS or higher in Computer Science or a related degree
• Familiarity with modern, Sparse-BEV Joint Detection and Tracking models.
• Hands-on experience with production ML pipelines, including dataset creation, metrics, and optimization
• Experience in Python, PyTorch, or a related ML framework

Bonus Qualifications

• Experience with CUDA code and/or TensorRT
• Strong mathematical skills and understanding of probabilistic techniques
• Experience implementing tracking systems

Base Salary Range

There are three major components to compensation for this position: salary, Amazon Restricted Stock Units (RSUs), and Zoox Stock Appreciation Rights. A sign-on bonus may be offered as part of the compensation package. The listed range applies only to the base salary. Compensation will vary based on geographic location and level. Leveling, as well as positioning within a level, is determined by a range of factors, including, but not limited to, a candidate's relevant years of experience, domain knowledge, and interview performance. The salary range listed in this posting is representative of the range of levels Zoox is considering for this position.

We may use artificial intelligence (AI) tools to support parts of the hiring process, such as reviewing applications, analyzing resumes, or assessing responses and identifying potential inconsistencies or verification signals in application materials based on available information. These tools assist our recruitment team but do not replace human judgment. Final hiring decisions are ultimately made by humans. If you would like more information about how your data is processed, please contact us.