Skip to content
ZooxZooxFoster City, CA

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 – 407k
HybridML Engineering

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
Zoox

Senior Staff Machine Learning Engineer - Perception

ZooxBoston, MA

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 – 389k
HybridML Engineering
Square

Staff Machine Learning Engineer, Underwriting and Credit

SquareUnited States

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 – 415k
Remote10+ YOEML Engineering
Square

Staff Applied Machine Learning Engineer - Fraud & Abuse

SquareSan Francisco, CA

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 – 415k
On-site12+ YOEML Engineering
Square

Staff Applied Machine Learning Engineer

SquareUnited States

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 – 415k
Remote12+ YOEML Engineering
Otter

Staff Machine Learning Engineer

OtterMountain View, CA

Lead architect and builder of large-scale ASR/NLP/LLM systems for Otter's conversational intelligence products. Owns end-to-end ML lifecycles from research to production deployment, mentoring engineers and setting technical direction.

278k – 330k
Hybrid10+ YOEML Engineering