← All roles

Staff Machine Learning Engineer - Deployment

Kodiak AI· AI & Machine Learning
Apply on company site
Hybrid (San Francisco Bay Area)$200k - $265kfull-timePosted 6mo ago

About the role

Staff ML Engineer focused on deploying onboard and offboard machine learning models for the Kodiak Driver autonomous system. Collaborates across perception, planning, simulation, and infrastructure to improve the autonomy stack using large-scale fleet data, with emphasis on deployment, quantization, pruning, ONNX/TensorRT conversion, and performance profiling.

Responsibilities

  • Help design, train, with a focus towards deploying onboard machine learning models that improve the performance/latency of the Kodiak autonomy stack which includes quantization, pruning, converting to ONNX/TensorRT, custom GPU kernels and profiling.

Qualifications

  • MS or PhD in Computer Science, Robotics, Machine Learning, or a related technical field, or equivalent practical experience.
  • 6+ years experience deploying machine learning or deep learning systems in production environments.
  • Proficiency in Python, C++ and modern ML frameworks such as PyTorch or TensorFlow.
  • Experience working with large-scale distributed training systems.
  • Solid understanding of optimization, and machine learning fundamentals.
  • Strong software engineering fundamentals including testing, debugging, and system design.
  • Ability to work collaboratively across teams to solve complex autonomy and robotics challenges.

Skills

Required:Machine Learning DeploymentPythonC++PyTorchTensorFlow
Nice to have:Distributed Training

Benefits

  • Competitive compensation package including equity and annual bonuses
  • Excellent Medical, Dental, and Vision plans through Kaiser Permanente, Cigna, and MetLife (including a medical plan with infertility benefits)
  • MetLife Legal Services, Identity & Fraud Protection, Hospital Indemnity Insurance, Accident Insurance, & Critical Illness Insurance
  • Flexible PTO, 10 paid holidays, and generous parental leave policies
  • Office located in Mountain View, CA
  • Dog-friendly office, free catered lunch, fully stocked kitchen, free EV charging
  • 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)
  • Fidelity 401(k)
  • Commuter, FSA, Dependent Care FSA, HSA
  • Various incentive programs (referral bonuses, patent bonuses, etc.)
Last updated Oct 21, 2026. Interested? Apply on the company site.

About Kodiak AI

Safely driving an autonomous future.

Our purpose-built autonomous solution, the Kodiak Driver, is the AV industry’s most advanced driverless technology, seamlessly integrating into any driving platform for reliable movement in any environment. Learn how we’re shaping the future of ground transportation.

autonomousdriverlessground transportationAI
Company profileWebsite

More open roles at Kodiak AI