Skip to content

Staff / Senior Staff Software Engineer - Learned Trajectory Machine Learning Engineer

Develops deep learning models using imitation and reinforcement learning to generate safe, efficient driving trajectories for autonomous vehicles. Collaborates with Perception, Planning, and Simulation teams; requires ML expertise, Python fluency, and transformer experience.

277k – 407kFoster City, CAML EngineeringHybrid

About the role

Responsibilities

  • Develop new deep learning models using imitation learning and reinforcement learning to generate driving plans for autonomous vehicles.
  • Work on techniques to estimate the quality of driving plans along dimensions of safety, progress, comfort, etc.
  • Leverage large-scale machine learning infrastructure to discover new solutions and push the boundaries of the field.
  • Develop metrics and tools to analyze errors and understand improvements of systems.
  • Collaborate with engineers on Perception, Planning, and Simulation to solve the overall Autonomous Driving problem in complex urban environments.

Qualifications

  • BS, MS, or PhD degree in computer science or related field.
  • Experience with training and deploying transformer-based model architectures and reinforcement learning.
  • Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines.
  • Fluency in Python with a basic understanding of C++.
  • Extensive experience with programming, algorithm design, and strong mathematics skills.

Bonus Qualifications

  • Conference or Journal publications in Machine Learning or Robotics related venues.
  • Prior experience with Prediction and/or autonomous vehicles or robotics in general.

Skills

PythonC++TransformersReinforcement LearningImitation LearningMachine Learning PipelinesDeep LearningMathematicsAutonomous DrivingRobotics

Similar roles

ML Engineering jobs

Staff Machine Learning Engineer - Mapping

Lead development of ML models and algorithms for HD 3D mapping in autonomous driving. Requires 5+ years experience, MS/PhD in CS, expertise in computer vision, deep learning, Python/C++, and cross-functional leadership.

277k – 333kFoster City, CAML EngineeringHybrid5+ YOEC++Slam

Senior Staff Machine Learning Engineer - Perception

Leads development of ML algorithms for robot perception, including scene understanding, tracking, segmentation, and multi-modal foundation models using sensor data. Requires deep expertise in deep learning, computer vision, and production ML pipelines, collaborating across autonomy teams.

277k – 389kBoston, MAML EngineeringHybridC++Lidar

Staff Machine Learning Engineer, Underwriting and Credit

Senior IC building and maintaining ML underwriting and credit decisioning models for Cash App Borrow and Afterpay. Owns full modeling lifecycle including experimentation, calibration, deployment, and monitoring.

277k – 415kUnited StatesML EngineeringRemote10+ YOESQLGCP

Staff Applied Machine Learning Engineer - Fraud & Abuse

Staff Applied ML Engineer building and operating production ML decision systems to detect and prevent payment fraud, scams, identity abuse, and marketplace risk across Block.

277k – 415kSan Francisco, CAML EngineeringOn-site12+ YOESQLJava

Staff Applied Machine Learning Engineer

Build and operate production ML systems for ranking, recommendations, search, and customer intelligence signals used across product, growth, risk, and decisioning teams. Requires 12+ years of production ML experience and deep expertise in intelligent systems.

277k – 415kUnited StatesML EngineeringRemote12+ YOESQLJava