Rivian is looking to develop the infrastructure to compile state-of-the-art machine learning models used in ADAS systems to execute efficiently on the SoC, enabling HW-SW codesign and efficient building blocks for state-of-the-art machine learning models.
Requirements
- Hands-on experience with quantized model deployment, ML compilation 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
- Prior experience working with hardware-software co-design, especially for autonomous or robotics platforms.
- Deep knowledge of numerical precision trade-offs, quantization-aware training (QAT), and dynamic/static quantization flows.
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.
Other
- 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.
- Equal Opportunity Employer: Rivian is an equal opportunity employer and complies with all applicable federal, state, and local fair employment practices laws.
- Candidate Data Privacy: Rivian may collect, use and disclose your personal information or personal data when you apply for employment and/or participate in our recruitment processes.
- Rivian is committed to ensuring that our hiring process is accessible for persons with disabilities.
- Please note that we are currently not accepting applications from third party application services.
