In this role, you will be a vital member of the ML Compiler team at Rivian, focusing on developing software tools for deep learning network inference on Rivian hardware platforms. You will collaborate closely with the Autonomy and Hardware teams to optimize performance and support new hardware features. The position emphasizes hardware-aware optimizations and efficient building blocks for machine learning models. Your work will contribute to the advancement of Rivian's autonomous vehicle technology.

• Ph.D. or M.S. in Computer Engineering or a related field.
• Excellent C/C++ and Python programming skills.
• Experience with various SOC platforms used for machine learning.
• Strong understanding of deep learning software models.
• Experience in compiler pipeline development preferred.
• Proficiency in deep learning frameworks and their low-level IRs or export formats.

• Lead the development of an ML Compiler for mapping Autonomy ML models to Rivian Autonomy Processor (RAP1).
• Design and implement hardware-aware optimizations, including quantization strategies and model compression.
• Collaborate with hardware teams to co-optimize model architecture and compute pipeline under real-time constraints.
• Benchmark and analyze system performance across platforms to achieve optimal deployment efficiency.
• Partner with autonomy teams to align model optimization efforts with hardware roadmap and real-world autonomy requirements.

• Employees at Rivian are often offered comprehensive health, dental, and vision insurance, a 401(k) with company match, ESPP and potential RSU grants, performance bonuses, and generous time off including paid parental and sick leave. Perks can include 24/7 mental-health coaching and therapy, Hinge Health for Anthem enrollees, tuition assistance and professional development, plus on-site food and a wide range of employee discounts on items like computers, mobile phones, home loans, and pet care.