Point your AI agent at freehire and let it find you a job.

Get the CLI →

Kodiak AI, Inc.

Staff Machine Learning Engineer Deployment

Posted Updated 2 views
Discussion

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.

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

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

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

Skills

What Staff ML / AI jobs ask for — and how much of it you have →
Apply

See also

ML / AI jobs by country — openings, pay and top skills →

Tailor your CV for this role?

We couldn't check your fit for this role — add a CV to your profile to see it next time.

A new version of freehire is available