Sr. Machine Learning/AI Engineer

Job Description

Rivian is seeking a Research Scientist with expertise in quantized deep learning models for hardware acceleration in autonomous systems.

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 and their optimization for edge inference
• Proficiency in deep learning frameworks and their low-level IRs or export formats
• Solid programming skills in C++, Python
• Familiarity with CUDA/OpenCL or other accelerator programming models

Responsibilities

• Research state of the art perception models
• Lead the development of optimizations for mapping quantized perception models to hardware platforms
• Design and implement hardware-aware optimizations for custom accelerators
• Collaborate with hardware teams to optimize model architecture and compute pipeline
• Benchmark and analyze system performance across platforms
• Partner with perception, systems, and autonomy teams

Benefits

• Robust medical/Rx, dental and vision insurance packages for full-time and part-time employees, their spouse or domestic partner, and children up to age 26.