JobsMachine Learning/AI Engineer
Rivian logo

Machine Learning/AI Engineer

Rivian

Location

Palo Alto, CA

Type

Full-time

Posted

5/9/2026

Compensation

$139,000 - $155,000 per year

Undergraduate with 2+ Years of Experience
Approval 99%·Filings 685·New hires 98·
Established Sponsor
·FY 2025

Job description

Rivian is seeking a Research Scientist with expertise in quantized deep learning models for hardware acceleration in autonomous systems. This role involves bridging perception model design and hardware-aware deployment to enable efficient execution of high-performance perception algorithms. The successful candidate will focus on researching state-of-the-art perception models and developing optimization pipelines for quantized versions tailored for real-time performance. Collaboration with various teams is essential to align model optimization efforts with hardware requirements.

Requirements

  • B.S. in Computer Engineering, Electrical Engineering, Computer Science, or related field with a focus on ML compilers, embedded systems, or hardware-aware AI.
  • Hands-on experience with quantized model deployment, ML design stacks, and code generation for embedded or heterogeneous compute systems.
  • Understanding of computer vision models such as object detection and segmentation and their optimization for edge inference.
  • Experience in deep learning frameworks like PyTorch or TensorFlow and their low-level IRs or export formats such as ONNX.
  • Solid programming skills in C++ and Python.
  • Familiarity with CUDA/OpenCL or other accelerator programming models.

Responsibilities

  • Research state-of-the-art perception models in collaboration with the ADAS SW teams.
  • Contribute to the development of optimizations for mapping quantized perception models to embedded and heterogeneous hardware platforms.
  • 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 perception, systems, and 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 posting expired or inaccurate?