Kodiak Robotics
Staff Machine Learning Engineer - Deployment
$200K–$265K
Mountain View, CA +1 · Staff · Full-time · Posted Apr 20, 2026
- Location
- Mountain View, CA
Mountain View, California, United States - Experience
- 6+ years deploying machine learning or deep learning systems in production environments.
- Education
- MS or PhD in Computer Science, Robotics, Machine Learning, or a related technical field, or equivalent practical experience.
- Visa sponsorship
- Visa sponsorship may be provided for eligible candidates.
About the job
Deploy and optimize machine learning systems for Kodiak’s autonomous driving stack, improving vehicle understanding, prediction, and safe driving decisions.
What you'll do
- Optimize and deploy onboard ML models, including quantization, pruning, ONNX/TensorRT conversion, GPU kernels, and profiling.
- Validate parity between onboard and offboard models using HIL, and integrate models with cross-functional production systems.
- Analyze driving logs and profiling results to improve model robustness, safety, and training performance; support scalable fleet deployment infrastructure.
What they ask for
- 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, plus software engineering fundamentals including testing, debugging, and system design.
Pay
$200,000 — $265,000 USD (base salary)
Tools
Python, C++, PyTorch, TensorFlow, ONNX, TensorRT, HIL
AI focus
Deploy and optimize onboard machine learning models for autonomous driving, including perception, prediction, planning, and continuous fleet learning.
Model-written summary · the original posting is authoritative