Software Engineer, ML Foundations Platform

Zoox is developing autonomous robotaxis to reimagine transportation with ground-up built vehicles that are safe, reliable, and clean.
$205,000 - $260,000
Machine Learning
Senior Software Engineer
Hybrid
6+ years of experience
AI · Automotive

Description For Software Engineer, ML Foundations Platform

Zoox is pioneering the future of transportation with their mission to develop fully autonomous robotaxis. As a Software Engineer on the ML Foundations Platform team, you'll play a crucial role in enabling ML innovations for autonomous driving. The position involves working across all ML teams within Zoox, including Perception, Behavior ML, Simulation, and Data Science.

You'll be responsible for building and operating the foundational ML tools, deep learning frameworks, and inference libraries used for both in-vehicle and off-vehicle ML applications. This role offers significant impact in reducing the time from ideation to production of cutting-edge AI innovations, particularly as Zoox expands its robotaxi deployments.

The position offers competitive compensation including base salary, Amazon RSUs, and Zoox Stock Appreciation Rights, along with comprehensive benefits. Working in a hybrid environment in Foster City, CA, you'll be part of a team that values diverse perspectives and is committed to building groundbreaking autonomous vehicle technology.

This is an excellent opportunity for experienced engineers who want to shape the future of autonomous transportation while working with cutting-edge ML technologies and frameworks. The role offers substantial growth opportunities as Zoox continues to expand its deployments and explore new ML domains.

Last updated 15 days ago

Responsibilities For Software Engineer, ML Foundations Platform

  • Design, develop, and deploy custom and off-the-shelf ML libraries and toolings
  • Build tooling and establish development best practices to manage and upgrade foundational libraries
  • Collaborate with cross-functional teams to define requirements and align on architectural decisions

Requirements For Software Engineer, ML Foundations Platform

Python
  • 6+ years of experience
  • Proficient in Python or C++
  • Familiarity with training frameworks like PyTorch, Lightning, Hugging Face, Ray, JAX
  • Familiarity with GPU-accelerated inference on Nvidia hardware like CUDA, TensorRT, and/or XLA

Benefits For Software Engineer, ML Foundations Platform

Medical Insurance
Dental Insurance
Vision Insurance
Equity
  • Paid time off (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

Interested in this job?

Jobs Related To Zoox Software Engineer, ML Foundations Platform

Senior Machine Learning Engineer - Collision Avoidance System

Senior Machine Learning Engineer position at Zoox focusing on collision avoidance systems for autonomous vehicles

Senior Software Development Engineer, Ring & Blink AI

Senior Software Engineer position at Amazon's Ring & Blink AI team focusing on computer vision and machine learning software development for smart home devices.

Senior Software Developer, Amazon Games AI

Senior Software Developer position at Amazon Games focusing on implementing ML, RL, and Generative AI techniques for game development.

Product Development Engineer, Annapurna Labs Silicon Operations

Senior Product Development Engineer role at AWS-Annapurna Labs focusing on silicon yield optimization for machine learning accelerator servers.

Sr. ES Product Manager

Lead AI and Agentforce Product Manager role at Salesforce, focusing on Employee Success products and solutions with 5+ years of product management experience required.