Senior ML Engineer, Perception

Rivian
Palo Alto, United States of America
6 days ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Languages
English
Experience level
Senior

Job location

Palo Alto, United States of America

Tech stack

Systems Engineering
Big Data
Python
Machine Learning
Information Technology
Machine Learning Operations
Lidar

Job description

Auto-labelling is a foundational pillar of the Autonomy stack. In this Senior ML Engineer role, you will play a key role in delivering high-quality, scalable auto-labeling models. This includes training, optimizing and shipping auto-labeling models in the Autonomy stack. Use cases include mapping, lanes auto-labelling, object auto-labelling as well as other critical applications. You will ship production-grade models that push the boundaries of what's possible. As such, you will also contribute to the whole end-to-end ML lifecycle & data flywheel of this effort: data acquisition, metrics definition, evaluation, model performance optimization, feedback loop. A key part of the role is especially dedicated to lidar-free auto-labeling, i.e. ship auto-labeling models that do not require lidar data.

  • Deliver prod-grade, high-quality, scalable auto-labeling models. Use cases include AV mapping, lanes auto-labelling and/or object auto-labelling, among other critical applications.
  • Train, optimize, ship auto-labeling models in the Autonomy stack, and continuously improve their performance.
  • Deliver auto-labeling with and without lidar data.
  • Establish rigorous evaluation and monitoring benchmarks. Identify and root-cause top-tier system anomalies, prioritizing high-impact optimizations to continuously push the needle on performance.
  • Partner closely with the Autonomy group to ensure we meet the feature requirements

Requirements

  • Education: BS, MS, or PhD in Computer Science, Robotics, Electrical Engineering, or a highly related quantitative field.
  • Experience: 5+ years of professional experience building and scaling ML solutions, with a strong focus on the following:
  • AV auto-labeling system at scale: Proven track record of hands-on experience delivering auto-labeling models for Autonomous Vehicles at scale. Auto labeling for mapping, lanes auto-labelling and/or object auto labelling.
  • Perception stack: solid understanding of the AV perception stack.
  • System engineering: Strong proficiency in Python alongside a solid understanding of modern Perception pipelines, benchmarking tools, and infrastructure.
  • Execution: Demonstrated ability to drive progress across a complex, multi-domain system, in a fast-paced environment., * 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

Apply for this position