Senior Machine Learning Engineer - Perception Mapping (AI)

TL;DR

Senior Machine Learning Engineer - Perception Mapping (AI): Designing, training, validating, and integrating ML models to detect semantic map elements for autonomous vehicles with an accent on data curation, model validation pipelines, and onboard integration. Focus on enabling scaling and autonomy capabilities by ensuring safe and efficient navigation in complex environments.

Location: Hybrid in Foster City, CA

Salary: $242,000–$290,000 a year

Company

Zoox is an autonomous vehicle technology company focused on creating self-driving robots for safe and efficient navigation.

What you will do

• Curate, validate, and label datasets for model training and validation.
• Research, implement, and train ML models to perform semantic map element detection.
• Closely collaborate with validation teams to formulate and execute model validation pipelines.
• Integrate models into the greater onboard autonomy system within compute budgets.
• Be a technical leader on the team, maintaining coding and ML development best practices and contributing to architectural decisions.

Requirements

• MS or PhD or equivalent experience (5+ years) in Computer Science or related field.
• Experience in computer vision or robotics.
• Experience with training and deploying deep learning models.
• Experience with Python libraries (pytorch, numpy).

Nice to have

• Experience with C++.
• Experience with CUDA and/or GPU programming.
• Experience with mapping related ML techniques.

Culture & Benefits

• Comprehensive compensation package including salary, Amazon Restricted Stock Units (RSUs), and Zoox Stock Appreciation Rights.
• Offers a package of benefits, including paid time off (sick leave, vacation, bereavement), and unpaid time off.
• Health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance.
• Uses artificial intelligence (AI) tools to support parts of the hiring process, but human judgment makes final hiring decisions.