Staff Machine Learning Engineer - Deployment

Mountain View, CAfull timeOn Site

Apply on Employer Site

senior

$200k–$265k

H1B: yes

Kodiak Staff Machine Learning Engineer focusing on model deployment to improve Kodiak Driver's autonomy stack performance and latency. Collaborate with cross-functional teams to integrate ML models into production and analyze real-world driving data for model robustness.

Responsibilities

This is a high-impact opportunity to work on cutting-edge AI systems operating in the real world, where every mile driven improves our models and brings autonomous trucking closer to global deployment.
Help design, train, with a focus towards deploying onboard machine learning models that improve the performance/latency of the Kodiak autonomy stack which includes quantization, pruning, converting to ONNX/TensorRT, custom GPU kernels and profiling.
Help 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 additionally help improve the train time for models by identifying and removing bottlenecks.
Contribute to the development of scalable AI infrastructure that supports continuous learning and deployment across Kodiak’s fleet.
What you’ll bring:
MS or PhD in Computer Science, Robotics, Machine Learning, or a related technical field, or equivalent practical experience.
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.

Qualification

MS or PhD in Computer ScienceProficiency in PythonSolid understanding of optimization

Required

MS or PhD in Computer Science, Robotics, Machine Learning, or a related technical field, or equivalent practical experience
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

Benefits

Medical
Dental
Vision
401(k)
PTO
Equity/RSU
Parental Leave
Life Insurance
Wellness

Company

Kodiak

View company page

Source: greenhouse

Active within the past hour

More roles at Kodiak

World Model Research Scientist- Physical AI

San Francisco Bay Area, United States

Vice President of Commercialization

Remote, United States

Technical Lead, Multimodal Transformers

Mountain View, CA

Staff Software Engineer, Mission Planning

Mountain View, CA

Boost Your Interview Chances

Improve Resume Match Score

Free

Your Score

9.0

Strong Applicant

Must-Have Skills for This Role

PythonC++PyTorchTensorFlowONNXTensorRT

Optimize my Resume

Apply Faster with Autofill

Free

Save your profile once. LumiJobs pre-fills the employer's application form so you stop re-typing the same answers on every site.

Apply With Autofill

Or skip LumiJobs's tools and submit directly on the employer's career site.

Apply on Employer Site

Staff Machine Learning Engineer - Deployment at Kodiak · LumiJobs