Staff Machine Learning Engineer - Deployment

Kodiak AI· Mountain View, CA

Posted October 27, 2026

Apply
Experience
6+ years
Pay
$200K – $265K per year
Education
MS or PhD in Computer Science, Robotics, Machine Learning, or equivalent
Location
Mountain View, CA
Work mode
On-site
Job type
Full-time

About this role

Kodiak AI builds vehicle-agnostic autonomous driving technology powering long-haul and industrial trucking. The machine learning deployment team optimizes and deploys onboard perception, prediction, and planning models onto autonomous trucks.

What you will do

  • Design and deploy onboard machine learning models optimizing latency and performance
  • Perform model quantization, pruning, ONNX or TensorRT conversion, and custom GPU profiling
  • Validate deployed offboard models against onboard performance using HIL testing
  • Collaborate with robotics and simulation teams to integrate ML models into production systems
  • Analyze real-world driving logs to identify edge cases and improve model robustness

What the employer expects

  • MS or PhD in Computer Science, Robotics, Machine Learning, or related field
  • 6+ years of experience deploying machine learning or deep learning systems in production
  • Proficiency in Python, C++, and modern ML frameworks like PyTorch or TensorFlow
  • Experience with large-scale distributed training and model optimization
  • Strong software engineering fundamentals in testing, debugging, and system design

Benefits

  • Competitive compensation including equity and annual bonuses
  • Comprehensive medical, dental, and vision insurance plans
  • Flexible PTO, 10 paid holidays, and parental leave
  • Office perks including catered lunch and EV charging

Skills

  • Python
  • C++
  • PyTorch
  • TensorFlow
  • ONNX
  • TensorRT
  • CUDA
  • Autonomous Driving

Apply on the employer's site

The application opens on Kodiak Greenhouse job board. Dev Portal Notes does not collect your application or charge a fee.

Apply now

Know someone who fits this role?

Applications are handled on the employer's own site.

Close the skill gap

Some links on this page may generate a small commission at no extra cost to you.

Similar jobs

Apply for this job