Staff Machine Learning Engineer Deployment
Kodiak AI is an active public company providing AI-powered autonomous ground-vehicle technology, principally through the Kodiak Driver for commercial trucking, industrial trucking, and defense applications.
Funding history
About Kodiak AI, Inc.
Kodiak Driver combines AI-powered software and modular, vehicle-agnostic hardware to enable driverless vehicle operations. Kodiak commercializes the technology through a Driver-as-a-Service model and serves trucking, industrial, and defense customers.
Skills
About the Role
You will design, train, optimize, and deploy onboard machine-learning models for autonomous driving. You will validate parity between onboard and offboard models, analyze driving data and edge cases, profile model and training performance, and integrate models into production autonomy systems.
Requirements
- Computer Science, Robotics, Machine Learning, or related technical degree or equivalent experience
- 6+ years deploying machine-learning or deep-learning systems in production
- Python and C++ proficiency
- PyTorch or TensorFlow experience
- Large-scale distributed training experience
- Machine-learning and optimization knowledge
- Software testing, debugging, and system-design skills
Responsibilities
- Design, train, and deploy onboard machine-learning models
- Optimize models through quantization, pruning, ONNX or TensorRT conversion, GPU kernels, and profiling
- Validate parity between onboard and offboard models using HIL
- Integrate ML models into production systems with cross-functional partners
- Analyze driving logs to identify edge cases and improve robustness and safety
- Profile training workloads and remove bottlenecks
- Develop scalable AI infrastructure for continuous learning and deployment
Benefits
- Equity and annual bonuses
- Medical, dental, and vision insurance
- Infertility benefits
- Legal services
- Identity and fraud protection
- Hospital indemnity, accident, and critical illness insurance
- Flexible PTO
- 10 paid holidays
- Parental leave
- Free catered lunch
- Free EV charging
- Long-term and short-term disability insurance
- Life insurance
- Wellbeing programs
- 401(k)
- Commuter benefits
- FSA, Dependent Care FSA, and HSA
- Referral and patent bonuses