Staff ML Engineer, Perception

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

Staff ML Engineer role focused on developing and shipping production-grade auto-labeling models for autonomous vehicles, with a specific emphasis on lidar-free solutions and the end-to-end ML lifecycle.

What you'd actually do

  1. Drive and deliver prod-grade, high-quality, scalable auto-labeling models.
  2. Push the performance of lidar-free auto-labeling.
  3. Establish rigorous evaluation and monitoring benchmarks.
  4. Identify and root-cause top-tier system anomalies, prioritizing high-impact optimizations to continuously push the needle on performance.
  5. Partner closely with the Autonomy group to ensure we meet the feature requirements

Skills

Required

  • BS, MS, or PhD in Computer Science, Robotics, Electrical Engineering, or a highly related quantitative field.
  • 7+ years of professional experience scaling ML solutions
  • Python
  • modern Perception pipelines
  • benchmarking tools
  • infrastructure
  • complex system spanning multiple domains and components

Nice to have

  • Experience in Lidar-free auto-labeling
  • Experience in mapping, especially from multiple vehicle passes and/or lidar-free mapping.
  • Experience in defining data annotation guidelines and partnering effectively with in-house and external 3P annotation vendors.
  • 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

  • Proven track record of hands-on experience driving and delivering auto-labeling models for Autonomous Vehicles at scale.
  • Auto labeling for mapping, lanes auto-labelling and/or object auto-labelling.
  • lidar-free auto-labeling

Other signals

  • shipping production-grade models
  • end-to-end ML lifecycle
  • lidar-free auto-labeling
  • perception stack