JobsMachine Learning/AI Engineer
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?
