Senior Machine Learning Engineer – ML Agents, Planning
Zoox
full-time
Posted on:
Location Type: Hybrid
Location: Foster City • California • 🇺🇸 United States
Visit company websiteSalary
💰 $242,000 - $290,000 per year
Job Level
Senior
Tech Stack
Python
About the role
- You will develop new deep learning models that use imitation learning and reinforcement learning to generate driving plans for human-like agents.
- You will work on novel techniques to estimate the quality of those driving plans along the dimensions of safety, progress, comfort and realism.
- You will contribute to our large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field
- You will develop metrics and tools to analyze errors and understand improvements of our systems
- You will collaborate with engineers on Perception, Planning, Simulation, and Validation to solve the overall Autonomous Driving problem.
Requirements
- PhD degree in computer science or related field +1y of professional experience (top tier publications can remove the need for the year of experience) or, MSc +5y of professional experience in a relevant field.
- Experience in Planning and / or Prediction using Reinforcement Learning techniques
- Experience with training and deploying transformer-based model architectures
- Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines
- Fluency in Python with a basic understanding of C++
Benefits
- paid time off (e.g. sick leave, vacation, bereavement)
- unpaid time off
- Zoox Stock Appreciation Rights
- Amazon RSUs
- health insurance
- long-term care insurance
- long-term and short-term disability insurance
- life insurance
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
deep learningimitation learningreinforcement learningtransformer-based model architecturesproduction machine learning pipelinesdataset creationtraining frameworksmetrics pipelinesPythonC++
Certifications
PhD in computer scienceMSc in relevant field