# Senior AI Infrastructure Engineer - Model Training at Kodiak

- Company: Kodiak
- What the company does: Backed by Battery, Lightspeed and CRV.
- Company website: http://www.kodiak.ai/
- Type: Startups (AI role)
- Level: Senior
- Location: Mountain View, CA
- Work setup: On-site
- Posted: 2026-07-09
- Apply by: 2026-10-08
- Apply: https://job-boards.greenhouse.io/kodiak/jobs/4310775009
- Page: https://www.1752.vc/careers/jobs/kodiak-senior-ai-infrastructure-engineer-model-training/

## About the role

Kodiak's AI is only as good as the speed at which we can train it. Every improvement to our models – from GigaFusionNet to large-scale world models – depends on infrastructure that turns thousands of hours of multimodal driving data into training throughput. We are looking for engineers who make model training fast: streaming massive camera, LiDAR, and radar datasets without stalling a single GPU, sharding data and models efficiently across nodes, and extracting every FLOP from the latest hardware.

## What they're looking for

- Design high-throughput data loading and streaming systems for multimodal sensor data (camera, LiDAR, radar), including dataset formats, sharding strategies, and prefetching pipelines that keep GPUs saturated
- Build and optimize distributed training infrastructure across multi-node GPU clusters, applying data, tensor, pipeline, and fully sharded (FSDP/ZeRO) parallelism to models that don't fit on a single device
- Maximize utilization of modern accelerators such as NVIDIA B200s through mixed-precision training (BF16/FP8), fused kernels, memory optimization, and communication/computation overlap
- Profile end-to-end training pipelines to find and eliminate bottlenecks across storage, network, CPU preprocessing, and GPU compute
- Develop scalable dataset construction pipelines that convert petabytes of raw driving logs into training-ready, streamable formats
- Partner with ML teams to scale new architectures from prototype to full-cluster training runs efficiently and reliably

Tags: AI & Machine Learning