Senior Machine Learning Engineer - Perception Mapping (AI)

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.