Skip to content

Software Engineer - Prediction and Behavior ML

125k – 222kSunnyvale, CAOnsite3+ YOE
Summary

Develops ML-first behavior prediction modules to forecast road user motions and interactions for autonomous systems. Requires 3+ years experience with deep learning end-to-end cycles, C++/Python fluency, and collaboration with perception/planning teams.

About the role

Responsibilities

  • Prototype, evaluate, refine, and deploy state-of-the-art ML algorithms that predict the future motion of the ego agent and other road users in the scene
  • Leverage established products at Applied Intuition to build the software and infra foundation for behavior-specific ML development
  • Collaborate tightly with perception and planning engineers on many cross-team initiatives

Requirements

  • Bachelor’s in Computer Science, Electrical Engineering, Robotics, or related field
  • Experience with the end-to-end development cycle of deep learning models
  • Expertise in subdomains such as modeling, input pipelines, evaluation, deployment, and model optimization
  • 3+ years of experience building production software using modern software practices
  • Fluency in C++, or fluency in Python with intermediate experience in C++

Nice to Have

  • Peer-reviewed research at machine learning conferences like NeurIPS, CVPR, ICML, ICLR, ICCV, ECCV, IROS, or ICRA
  • Experience with driver assistance or autonomous driving systems
  • Experience in evaluating and improving system-in-the-loop model performance

Compensation

Base salary range: $125,000 - $222,000 USD annually (plus equity and benefits)

Skills
Machine LearningDeep LearningC++PythonBehavior PredictionMotion ForecastingModel DeploymentModel OptimizationAutonomous DrivingPerception Systems
Similar roles at this salary range
All ML Engineering jobs →
Chime

AI/ML Engineer

Build and productionize ML models for risk detection and decisioning systems. Requires 1-2 years applied ML experience and familiarity with AWS, model evaluation, and experimentation.

125k – 173kSan Francisco, CAML EngineeringHybridAWSPython
PrizePicks

Machine Learning Platform Engineer

Build and operate the ML platform to productionize models, enable real-time inference, and manage the full ML lifecycle with MLOps best practices. Requires 3+ years platform engineering and 1+ years owning ML systems end-to-end.

135k – 160kUnited StatesML EngineeringRemoteGoKafka
Ai2

Senior Research Engineer

Research Engineer training and scaling flagship open multimodal and agentic models (Olmo, Molmo). Owns end-to-end ML infrastructure, model development, and open-source releases.

147k – 220kSeattle, WAML EngineeringOn-siteJAXMoE
Clickhouse

AI Product Engineer

Build agentic capabilities on a petabyte-scale observability platform. Own the full agent stack including context engineering, tool design, evals, and production reliability for incident investigation.

130k – 230kUnited StatesML EngineeringRemoteMCPSQL
Kodex

ML Engineer

Build and deploy production ML models and pipelines to detect suspicious activity, improve verification accuracy, and support threat intelligence workflows.

150k – 180kUnited StatesML EngineeringRemoteAWSClustering