Staff Software Engineer, AI Libraries
Core
Design and build high-quality, reusable AI libraries and platforms to support machine learning teams in training, evaluating, and deploying models at scale for autonomous driving.
Role type
Staff Software Engineer (AI Platform/Libraries)
Builds
Modular libraries for multimodal data loading, distributed training, inference, checkpointing, and observability.
Domain
Autonomous driving / AI Infrastructure
Deliverable
production ML models
Required skills
Python, distributed computing, software architecture, API design, testing, observability, project leadership
Preferred skills
PyTorch, TensorFlow, Azure, concurrent/parallel computing
Technologies
Python, PyTorch, TensorFlow, Azure
Responsibilities
Design and write Python code for scalable AI libraries, lead projects from discovery to delivery, improve engineering standards, support ML teams in scaling model performance, define technical direction for software architecture.
Seniority
Staff, hands-on IC with strategic influence
