Staff ML Engineer, Hardware Software Co-Design

Rivian is an automotive company focused on sustainable transportation and electric vehicles. They are seeking a Staff ML Engineer to bridge perception model design and hardware-aware deployment for autonomous systems, focusing on optimizing high-performance perception algorithms for embedded platforms.

Responsibilities:

• Research state of the art perception models in collaboration with the ADAS SW teams
• Lead the development of optimizations for mapping quantized perception models (e.g., CNNs, Transformers, LLMs) to embedded and heterogeneous hardware platforms
• Design and implement hardware-aware optimizations, including quantization strategies, model compression, memory-efficient representations, and operator fusion, targeted to custom accelerators
• Collaborate with hardware teams to co-optimize model architecture and compute pipeline under real-time constraints (latency, throughput, power)
• Benchmark and analyze system performance across platforms and iterate to achieve optimal deployment efficiency
• Partner with perception, systems, and autonomy teams to align model optimization efforts with hardware roadmap and real-world autonomy requirements