Kodiak Robotics
Senior AI Infrastructure Engineer - Model Training
$190K–$260K
Mountain View, CA +1 · Senior · Full-time · Posted Jul 9, 2026
- Location
- Mountain View, CA
Mountain View, California, United States - Experience
- At least 2-3 years of industry experience in ML systems or infrastructure.
- Education
- BS, MS, or PhD in Computer Science or a related field.
- Visa sponsorship
- Visa sponsorship may be provided for eligible candidates.
About the job
Build and optimize high-throughput data and distributed training infrastructure for multimodal autonomous-driving AI models.
What you'll do
- Design data-loading, streaming, and dataset-construction pipelines for large-scale camera, LiDAR, and radar data.
- Build and optimize multi-node GPU training, including parallelism, accelerator performance, and pipeline profiling.
- Partner with ML teams to scale model architectures from prototypes to full-cluster training.
What they ask for
- BS, MS, or PhD in Computer Science or a related field.
- At least 2-3 years of industry experience in ML systems or infrastructure.
- Hands-on experience with distributed training frameworks and techniques, and strong understanding of parallelism trade-offs.
- Experience building large-scale training data pipelines and deep understanding of GPU performance.
Nice to have
- Systems-level experience with C++/CUDA/Triton.
Pay
$190,000 — $260,000 USD (base salary)
Tools
PyTorch DDP/FSDP, DeepSpeed, Megatron, NCCL, WebDataset, MosaicML Streaming/MDS, NVIDIA B200s, Nsight
AI focus
Infrastructure for training autonomous-driving AI models using multimodal driving data, including GigaFusionNet and large-scale world models.
Model-written summary · the original posting is authoritative