Senior AI Infrastructure Engineer - Model Training

What this role involves

Kodiak Robotics is seeking a Senior AI Infrastructure Engineer to build and optimize distributed training infrastructure for autonomous driving. The role involves designing high-throughput data loading systems for multimodal sensor data, maximizing GPU utilization, and scaling new architectures from prototype to full-cluster training. The engineer will work closely with ML teams to improve model training speed.

Skills this posting asks for

• pytorch
• distributed training
• cuda
• nvidia gpu
• data pipelines
• multimodal data
• deep learning
• python
• mixed precision training
• gpu optimization
• fsdp
• deepspeed
• megatron
• webdataset
• mosaicml streaming
• lidar
• radar
• camera
• sensor fusion
• datasets
• profiling
• performance optimization
• ncc

Requirements

• 2 years of experience
• Level: senior

From the employer’s posting

Kodiak Robotics, Inc. was founded in 2018 and has become a leader in autonomous ground transportation committed to a safer and more efficient future for all. The company has developed an artificial intelligence (AI) powered technology stack purpose-built for commercial…