Kodiak Robotics

Kodiak Robotics

Senior AI Infrastructure Engineer - Model Training

Mountain View, CA · Senior

Sponsorship not specified$190k-$260kDetected 1 day ago
PythonC++Node.jsSpringAWSMachine LearningPyTorchData EngineeringIntellectual PropertySensorsCommunication

Stay score

odds of building a lasting career here

56Sponsors, lottery-bound
Cap-exempt (no lottery)0
Sponsors this role70
Entry-level history0
PERM / green-card track0
Lottery odds (Level IV)94
Fits your clock70

Sponsors, but it's cap-subject — you still face the weighted lottery (~61% per draw at Level IV). Good if you win; have a cap-exempt backup on your list.

Lottery odds assume a STEM candidate.

Personalize to your clock →

Community outcomes

No reports yet — be the first to help the next applicant.

About the role

  • Its Kodiak Driver is a vehicle-agnostic autonomous driving system that combines AI-powered software with modular hardware.
  • Today it operates in long-haul trucking, industrial trucking, and defense.
  • In 2024, Kodiak launched commercial Driver-as-a-Service operations with no one in the cab, and plans to launch driverless long-haul service between Dallas and Houston by the end of 2026.

Responsibilities

  • 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
  • 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
  • Experience building high-performance data pipelines for large-scale training, including streaming dataset formats (WebDataset, MosaicML Streaming/MDS, or similar), sharding, and storage/network-aware loading
  • Passion for building the infrastructure that lets AI for the physical world train faster, scale further, and improve continuously
  • At Kodiak, we strive to build a diverse community working towards our common company goals in a safe and collaborative environment where harassment of any kind is strictly prohibited.

Requirements

  • Hands-on experience with distributed training frameworks and techniques (PyTorch DDP/FSDP, DeepSpeed, Megatron, NCCL) and a strong grasp of parallelism trade-offs

Nice to have

  • Strong Python skills and proficiency in PyTorch internals
  • systems-level experience (C++/CUDA/Triton) a plus

Compensation

  • Competitive compensation package including equity and annual bonuses
  • California Pay Range
  • $190,000 - $260,000 USD

Benefits

  • Long Term Disability, Short Term Disability, Life Insurance
  • Wellbeing Benefits - Headspace through Cigna, Calm through Kaiser, One Medical, Gympass, Spring Health through Cigna, Rula (mental health navigation)
  • Commuter, FSA, Dependent Care FSA, HSA
  • Total compensation at Kodiak includes base pay, equity, bonus and a competitive benefits package
  • Actual starting pay will be based on job-related factors including: work location, experience, relevant training, education, skill level and performance during interview.

Company info

  • Kodiak AI, Inc. (Nasdaq: KDK), founded in 2018, is a leader in Physical AI, developing driverless technology that powers machines that move.
  • Kodiak has partnered with many of the biggest names in freight, including Bosch for scaled hardware production, AWS, and leading customers including Atlas Energy Solutions, IKEA, Bridgestone, Werner Enterprises, C.R. England, Roehl Transport, and General Dynamics Land Systems.
  • 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.
  • If you measure your impact in tokens per second and GPU utilization, this role is for you.
  • As part of the evaluation process, we provide Endorsed with job requirements and candidate-submitted applications.

Equal opportunity

  • equal opportunity employment regardless of race, ethnicity, religion, gender identity, sexual orientation, age, disability, or veteran status, or any other basis protected by applicable law.

Visa & Work Authorization

  • Should the position require, and Kodiak determines that a candidate's residence, U.S. person status, and/or citizenship status necessitate an export license, bar the candidate from the position, or otherwise fall under national security-rel
Apply directly at Kodiak Robotics →Create a free account for alerts like thisView Kodiak Robotics immigration profile

This listing is sourced directly from Kodiak Robotics's careers page and normalized into a canonical job model.