Staff Machine Learning Engineer - Deployment

Sourced from careers4graduates.com

Kodiak Robotics is seeking a Staff Machine Learning Engineer focused on model deployment to help build the intelligence that powers the Kodiak Driver. You will design and deploy machine learning systems that improve the vehicles' ability to understand the world, predict the behavior of other road users, and make safe driving decisions. Collaborate closely with robotics, autonomy, and infrastructure teams to continuously improve the performance of the autonomy stack using large-scale data from the growing fleet.

• Design, train, and deploy onboard machine learning models to improve performance and latency of the Kodiak autonomy stack, including quantization, pruning, conversion to ONNX/TensorRT, custom GPU kernels, and profiling.
• Identify and achieve parity between onboard and deployed offboard models using validation on HIL.
• Collaborate with cross-functional teams including perception, planning, simulation, and autonomy infrastructure to integrate ML models into production systems.
• Analyze real-world driving logs to identify edge cases and improve model robustness and safety.
• Use profiling tooling to improve model training time by identifying and removing bottlenecks.
• Contribute to the development of scalable AI infrastructure that supports continuous learning and deployment across Kodiak's fleet.