Senior AI Infrastructure Engineer Model Training
Kodiak AI is an active public company providing AI-powered autonomous ground-vehicle technology, principally through the Kodiak Driver for commercial trucking, industrial trucking, and defense applications.
Funding history
About Kodiak AI, Inc.
Kodiak Driver combines AI-powered software and modular, vehicle-agnostic hardware to enable driverless vehicle operations. Kodiak commercializes the technology through a Driver-as-a-Service model and serves trucking, industrial, and defense customers.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will build high-throughput sensor-data pipelines and distributed GPU training infrastructure. You will optimize parallelism, accelerator utilization, storage, networking, preprocessing, and compute; construct scalable datasets; profile bottlenecks; and scale model architectures to full-cluster training.
Requirements
- BS, MS, or PhD in computer science or a related field
- 2-3 years of industry experience in ML systems or infrastructure
- Experience with distributed training frameworks and techniques
- Experience building high-performance large-scale training data pipelines
- Understanding of GPU performance, profiling, and interconnects
- Strong Python and PyTorch skills
Responsibilities
- Design data-loading and streaming systems for multimodal sensor data
- Build and optimize distributed multi-node GPU training infrastructure
- Optimize accelerator utilization through mixed precision, kernel fusion, and memory optimization
- Profile training pipelines and eliminate bottlenecks
- Develop scalable driving-log dataset construction pipelines
- Scale new model architectures to full-cluster training runs
Benefits
- Equity
- Annual bonuses
- Medical, dental, and vision plans
- Infertility benefits
- Legal services
- Identity and fraud protection
- Hospital indemnity, accident, and critical illness insurance
- Flexible PTO
- 10 paid holidays
- Parental leave
- Dog-friendly office
- Free catered lunch
- Stocked kitchen
- Free EV charging
- Long-term and short-term disability insurance
- Life insurance
- Wellbeing programs
- 401(k)
- Commuter benefits
- FSA
- Dependent Care FSA
- HSA
- Referral and patent bonuses