Senior Deep Learning Sofware Infrastructure Engineer

Overview

Senior Deep Learning Software Infrastructure Engineer to build and scale training libraries and infrastructure for end-to-end autonomous driving models.

What you'll do

• Craft, scale, and harden deep learning infrastructure libraries and frameworks for training on multi-thousand GPU clusters.
• Improve efficiency across the training stack, including data loaders, distributed training, scheduling, and performance monitoring.
• Build robust training pipelines and libraries for massive video datasets and rapid experimentation.
• Collaborate with researchers, model engineers, and platform teams to minimize stalls and improve training availability.
• Own core infrastructure components such as orchestration libraries, distributed training frameworks, and fault-resilient training systems.

What you'll need

• Degree in CS, Electrical/Computer Engineering, or related field (or equivalent experience).
• 12+ years of professional experience building and scaling high-performance distributed systems.
• Extensive knowledge of deep learning frameworks (PyTorch preferred).
• Knowledge of large-scale training (DDP/FSDP, NCCL, tensor/pipeline parallelism).
• Experience with performance profiling.
• Strong systems background including networking (RoCE/IB) and storage/schedulers (Lustre, Slurm, Kubernetes, etc.).
• Proficiency in Python for production-grade libraries and orchestration/automation tools.

Details

• Remote work location listed as US, CA.