[ engineering ]

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 ]

Browse all open engineering roles →

[ 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.