Senior/Staff Software Engineer - Learned Trajectory Machine Learning Engineer

Zoox

over 1 year ago

Foster City, CA, USA

Senior / Staff+

H1B Sponsor

Base Salary

$230k - $332k/yr

Responsibilities

Develop new deep learning models for generating driving plans using imitation and reinforcement learning.
Estimate the quality of driving plans based on safety, progress, and comfort.
Leverage large-scale machine learning infrastructure to discover innovative solutions.
Develop metrics and tools to analyze errors and track system improvements.
Collaborate with engineers on Perception, Planning, and Simulation to address autonomous driving challenges.

BS, MS, or PhD degree in computer science or a related field.
Experience with training and deploying transformer-based model architectures and reinforcement learning.
Experience with production machine learning pipelines, including dataset creation and training frameworks.
Fluency in C++ or Python with a basic understanding of C++.
Extensive experience in programming and algorithm design.
Strong mathematics skills.

Comprehensive benefits package including paid time off, health insurance, and long-term care insurance.
Zoox Stock Appreciation Rights and Amazon Restricted Stock Units (RSUs).
Sign-on bonus may be included in the compensation package.