Staff Machine Learning Compiler Engineer

Rivian

Palo Alto, CA Full-time 3/17/2026

PhD Entry-Level

Approval 99%•Total filings 685•New hires 98•

✓ Established Sponsor

•FY 2025

Job Description

This position involves working as a key member of the ML Compiler team to develop software tools for deep learning inference on Rivian hardware platforms. The role includes collaborating with various teams to optimize model architectures and compute pipelines while ensuring performance under real-time constraints.

Requirements

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.
Experience working in aggressive design environments is preferred.

Responsibilities

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, model compression, memory-efficient representations, and operator fusion, targeted to RAP1.
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 autonomy teams to align model optimization efforts with hardware roadmap and real-world autonomy requirements.

Benefits

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.

Is this job posting expired or no longer available?