Senior Machine Learning Engineer (AI)

TL;DR

Senior Machine Learning Engineer (AI): Designing and implementing cutting-edge detection models using fused sensor data for the next generation of driving software. Focus on developing generalizable and adaptable models, joint detection and tracking models, and relaying semantic meaning to downstream models to capture uncertainty.

Location: Foster City, CA or Boston, MA

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

Company

Zoox is working to deliver safe and reliable autonomous mobility.

What you will do

• Train ML models and incorporate SOTA approaches, including ablations/modifications of existing models.
• Document experimental outcomes and drive the deployment of models into Zoox robots.
• Visualize data at the input and output to understand potential issues.
• Collaborate with teams such as ML optimization, data labeling, metrics, and planners to ensure project success.

Requirements

• MS/PhD in Computer Science or a related field.
• Experience with PyTorch or related ML frameworks.
• Experience deploying learned models into production.
• Excellent collaboration and teamwork skills, with a proven track record of working on large-scale problems.

Nice to have

• Prior experience in the AV or related space with joint detection and tracking models.
• Deep understanding of how to set up end-to-end training and evaluation processes.

Culture & Benefits

• Comprehensive package of benefits including paid time off.
• Includes Zoox Stock Appreciation Rights and Amazon RSUs.
• Health insurance, long-term care insurance, and life insurance provided.