Senior Ai Infrastructure Engineer - Model Training
Kodiak Robotics- Location
- Mountain View, California
- Work mode
- Remote
- Type
- full time
- Listed
Applications go directly to Kodiak Robotics. Ramai takes no fee for placement and no money for ranking.
[ About this role ]
Senior AI Infrastructure Engineer - Model Training at Kodiak Robotics. This role focuses on designing and optimizing high-throughput data loading and streaming systems for multimodal sensor data to keep GPUs saturated during model training. The engineer will build and optimize distributed training infrastructure across multi-node GPU clusters, applying various parallelism techniques.
Skills and focus areas
- data loading
- data streaming
- multimodal sensor data
- dataset formats
- sharding strategies
- prefetching pipelines
- distributed training
- gpu clusters
- data parallelism
- tensor parallelism
- pipeline parallelism
- fully sharded data parallelism (fsdp)
Summary written by Ramai from the original posting. Full details and requirements are on Kodiak Robotics’s careers page.
[ More at Kodiak Robotics ]
See all 67 open roles at Kodiak Robotics →[ Similar roles at other startups ]
- Mechanical Hardware EngineerSeeing Systems
- Forward Deployed EngineerGreenlite
- Founding Product EngineerVOYGR
- Software Engineer (toronto)Strada
- Member of Technical Staff (models)Jarmin
- Product Engineer (full-stack)Karumi
[ Not quite right? ]
Tell the assistant what you’re after — stage, location, stack, visa sponsorship — and it searches every open role for you. A few questions are free, no account needed.