Senior AI Infrastructure Engineer - Model Training
Kodiak · Mountain View, CA
Posted about 3 months ago
or apply directly on Kodiak's site. We never take the application ourselves.
Is this posting real?
- This role has been open
- 80 days Kodiak's roles stay open a median of 80 days
- Reposted
- No
- Salary listed
- No 0% of Kodiak's roles list one
- Ghost-job risk at Kodiak
- high 72 stale, 2 reposted of 72 open
- Hiring momentum
- 92 roles opened in the last 90 days ↑ up vs. the prior 90 days
- Last confirmed on the employer's board
- 2026-10-22
Measured from postings appearing on and disappearing from Kodiak's own greenhouse board since 2026-08-03. Full hiring picture for Kodiak.
About this role
As a Senior AI Infrastructure Engineer at Kodiak AI, you will focus on designing and optimizing high-throughput data loading and streaming systems for multimodal sensor data, ensuring efficient GPU utilization during model training. Your responsibilities will include building distributed training infrastructure, profiling training pipelines to eliminate bottlenecks, and collaborating with machine learning teams to scale new architectures from prototypes to full training runs.
- benefits
- 5/5
- freshness
- 1/5
- career value
- 5/5
- role clarity
- 5/5
- pay transparency
- 0/5
Scored from the posting itself — how clearly the role is described, how much it says about pay and benefits, and how recently it was listed. Not a judgement of Kodiak as an employer.
What you need
- BS, MS, or PhD in Computer Science or a related field
- 2-3 years of industry experience in ML systems or infrastructure
- Hands-on experience with distributed training frameworks (PyTorch DDP/FSDP, DeepSpeed, Megatron, NCCL)
- Experience building high-performance data pipelines for large-scale training
- Deep understanding of GPU performance including mixed precision and profiling tools
- Strong Python skills and proficiency in PyTorch internals
Nice to have
- Systems-level experience (C++/CUDA/Triton)
- Passion for building infrastructure for AI training
What you get
- Competitive compensation package including equity and annual bonuses
- Excellent Medical, Dental, and Vision plans
- Flexible PTO, 10 paid holidays, and generous parental leave policies
- Office perks: dog-friendly, free catered lunch, fully stocked kitchen, free EV charging
- Long Term Disability, Short Term Disability, Life Insurance
- Wellbeing Benefits - Headspace, Calm, One Medical, Gympass, Spring Health, Rula
Worth weighing
- No specific mention of remote work options
- The role involves working with complex distributed systems which may require a steep learning curve
- The company uses a third-party AI tool for initial application screening, which may affect the application process
Summarised from Kodiak's posting. Read the full original.
Listed by Kodiak on their greenhouse job board, last confirmed open on 2026-10-22. PitchMeAI is not the employer.
More roles at Kodiak
- Sr Product Manager, Assisted AutonomyMountain View, CA
- Senior HelpdeskMountain View, CA
- Senior Software Engineer, Behavior & Motion PlanningMountain View, CA
- Class A CDL DriverLancaster, Texas
- Vehicle Electrical Integration TechnicianOdessa, Tx
- Staff Software Engineer, Learned & Hybrid Behavior PlanningMountain View, CA
- Staff Autonomy Engineer, PerceptionMountain View, CA
- Sr Product Manager, Command and Field OperationsLancaster, TX