Kodiak

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.

Our read on this posting3.2out of 5
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

All 72 open roles at Kodiak →