Staff Machine Learning/AI Engineer
Rivian Palo Alto, CA Full-time 3/24/2026 $228k - $285k per year
PhD Entry-Level
Approval 99%•Total filings 685•New hires 98•
✓ Established Sponsor
•FY 2025Job Description
Rivian is seeking an Engineer / Research Scientist with expertise in quantized deep learning models for hardware acceleration in autonomous systems. The role involves researching perception models, optimizing them for embedded platforms, and collaborating with hardware teams to ensure efficient execution of algorithms.
Requirements
- Ph.D. or M.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.
- Strong understanding of computer vision models (e.g., object detection, segmentation) and their optimization for edge inference.
- Proficiency in deep learning frameworks (e.g., PyTorch, TensorFlow) and their low-level IRs or export formats (e.g., ONNX).
- Solid programming skills in C++, 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.
- 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.
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?