Senior ML Engineer - Deploy & Optimize Onboard Autonomy

Kodiak; San Francisco (CA); On-site; USD 200,000 - 265,000; Full time

Benefits offered by this job

Competitive compensation

Health + dental + vision

Flexible PTO

Dog-friendly office

401(k)

Job summary

Kodiak Robotics, Inc. is seeking a Senior Machine Learning Engineer focused on deploying models for the Kodiak Driver. You will design and deploy ML systems, optimize onboard latency, and collaborate with perception, planning, and autonomy infrastructure teams using real-world driving data.

The role emphasizes scalable AI infrastructure, rigorous testing, and continuous improvement across Kodiak's autonomous trucking platform in Mountain View, CA.

Qualifications

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

Responsibilities

• Design and deploy ML systems to improve onboard perception, prediction, and planning for autonomous trucking.
• Collaborate with robotics, autonomy, and infrastructure teams to productionize models.
• Analyze real-world driving data to improve model robustness and safety.
• Use profiling to reduce train time and latency of the autonomy stack.
• Ensure parity between onboard and offboard models via validation on HIL.
• Contribute to scalable AI infrastructure for continuous learning across Kodiak fleet.

Skills

Python

C++

PyTorch

TensorFlow

Distributed training

Software engineering

Education

MS or PhD in CS/Robotics/ML

Tools

ONNX/TensorRT

Profiling tools

Job description

Kodiak Robotics, Inc. is seeking a Senior Machine Learning Engineer focused on deploying models for the Kodiak Driver. You will design and deploy ML systems, optimize onboard latency, and collaborate with perception, planning, and autonomy infrastructure teams using real-world driving data.

The role emphasizes scalable AI infrastructure, rigorous testing, and continuous improvement across Kodiak's autonomous trucking platform in Mountain View, CA.