Software Engineer
Core
Building stable, scalable, and modular Python libraries, tools, and platforms to enable ML engineers and researchers to train, evaluate, and scale large-scale models for autonomous driving.
Role type
Senior IC software engineer (ML infrastructure/platform)
Builds
Scalable Python libraries, distributed training tools, and training infrastructure for ML teams
Domain
Autonomous driving / Machine Learning infrastructure
Deliverable
production ML models | infrastructure
Required skills
Python, software architecture, system design, cloud environments (Azure), concurrent/parallel/distributed computing, ML frameworks (PyTorch, TensorFlow, PyTorch Lightning)
Preferred skills
Large GPU clusters, distributed training (DDP, FSDP), observability (Prometheus, Grafana, Datadog, OpenTelemetry), data pipeline orchestration (Airflow, Flyte, Ray, Metaflow, Argo Workflows), containerization (Docker, Kubernetes, Terraform), ML system profiling (NVIDIA Nsight)
Technologies
Python, Azure, PyTorch, TensorFlow, PyTorch Lightning, DDP, FSDP, Prometheus, Grafana, Datadog, OpenTelemetry, Airflow, Flyte, Ray, Metaflow, Argo Workflows, Docker, Kubernetes, Terraform, NVIDIA Nsight
Responsibilities
Design and maintain scalable Python libraries and tools for ML workflows; Develop abstractions for data loading, distributed training, inference, and model evaluation; Support training at scale across large GPU clusters; Collaborate with ML teams to create reliable, well-documented, and observable tools; Improve engineering quality through testing, monitoring, and maintainability practices; Optimize data and training pipelines for multi-modal sensor data
Seniority
Senior, hands-on IC
