Senior Software Engineer, ML Core

Core responsibilities:

• Design, develop, and deploy custom and off-the-shelf ML libraries and toolings to improve ML development, training, deployment, and on-vehicle model inference latency.
• Build tooling and establish development best practices to manage and upgrade foundational libraries (e.g., Nvidia driver, PyTorch, TensorRT) to improve ML developer experience and expedite debugging efforts.
• Collaborate closely with cross-functional teams including applied ML research, high-performance compute, advanced hardware engineering, and data science to define requirements and align on architectural decisions.

This role enables innovations in ML and AI to make autonomous driving as seamless as possible, contributing to the development of safe, reliable, clean, and enjoyable autonomous robotaxis that will reimagine transportation for everyone.

Required qualifications:

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

No specific benefits, compensation, or salary information mentioned in the job posting.

Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market, aiming to provide the next generation of mobility-as-a-service in urban environments through robotics, machine learning, and design.