Senior AI Infrastructure Engineer Model Training
Public autonomous-vehicle technology company providing the Kodiak Driver, an AI-powered, modular autonomy platform for driverless commercial trucking and public-sector ground applications.
Funding history
About Kodiak AI, Inc.
Kodiak AI commercializes driverless ground-autonomy technology. Its vehicle-agnostic Kodiak Driver combines AI software, modular hardware, SensorPods, remote support, and offboard services; it serves trucking, industrial, defense, and public-sector use cases.
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