Staff Machine Learning Engineer - Deployment
Core
Design and deploy machine learning systems to improve autonomous trucking vehicles' ability to understand the world, predict road user behavior, and make safe driving decisions.
Role type
Staff Machine Learning Engineer (Deployment)
Builds
Onboard ML models for the Kodiak Driver autonomy stack
Domain
Autonomous trucking / Robotics
Deliverable
production ML models
Required skills
Python, C++, PyTorch, TensorFlow, distributed training, model optimization (quantization, pruning), ONNX/TensorRT, custom GPU kernels, profiling, system design
Preferred skills
HIL validation, edge case analysis, scalable AI infrastructure development
Responsibilities
Design and deploy onboard ML models to improve performance/latency; Achieve parity between onboard and offboard models via HIL validation; Integrate ML models into production systems with cross-functional teams; Analyze driving logs to identify edge cases and improve robustness; Profile models to reduce train time; Contribute to scalable AI infrastructure for continuous learning.
Seniority
Staff, hands-on IC with mentorship
