Senior ML Engineer, Perception

Rivian Rivian · Auto · Palo Alto, CA · Autonomous Driving

Senior ML Engineer focused on developing and scaling auto-labeling models for autonomous vehicles, including lidar-free applications. The role involves the full ML lifecycle from data acquisition to model optimization and deployment, with a strong emphasis on evaluation and system performance.

What you'd actually do

  1. Deliver prod-grade, high-quality, scalable auto-labeling models.
  2. Train, optimize, ship auto-labeling models in the Autonomy stack, and continuously improve their performance.
  3. Deliver auto-labeling with and without lidar data.
  4. Establish rigorous evaluation and monitoring benchmarks.
  5. Identify and root-cause top-tier system anomalies, prioritizing high-impact optimizations to continuously push the needle on performance.

Skills

Required

  • BS, MS, or PhD in Computer Science, Robotics, Electrical Engineering, or a highly related quantitative field.
  • 5+ years of professional experience building and scaling ML solutions
  • Proven track record of hands-on experience delivering auto-labeling models for Autonomous Vehicles at scale.
  • solid understanding of the AV perception stack.
  • Strong proficiency in Python
  • solid understanding of modern Perception pipelines, benchmarking tools, and infrastructure.
  • Demonstrated ability to drive progress across a complex, multi-domain system, in a fast-paced environment.

Nice to have

  • Experience in one of the following auto-labeling applications: mapping, lanes auto-labelling or object auto-labelling.
  • Experience in Lidar-free auto-labeling
  • Experience in mapping, especially from multiple vehicle passes and/or lidar-free mapping.
  • Experience in complex,multi-modal, large-scale data flywheel
  • Experience with multiple modalities (e.g., cameras, LiDAR, Radar).
  • Experience with onboard edge deployment, cloud inference architectures, and balancing compute/efficiency trade-offs

What the JD emphasized

  • AV auto-labeling system at scale
  • auto-labeling models for Autonomous Vehicles at scale
  • lidar-free auto-labeling
  • multi-modal, large-scale data flywheel

Other signals

  • delivering high-quality, scalable auto-labeling models
  • ship production-grade models
  • end-to-end ML lifecycle & data flywheel
  • lidar-free auto-labeling
  • Establish rigorous evaluation and monitoring benchmarks