Staff Software Engineer, Machine Learning Performance

Requirements

Must have:

- Masters or Ph.D. in Computer Science, Electrical, Mechanical, Aerospace Engineering, or a related discipline.

• Profound understanding of deep learning principles, heterogeneous computing, ML accelerators, and compilers with over 7 years of relevant industry experience.
• Proven team collaborator with exceptional communication skills to facilitate cross-functional initiatives in a dynamic development atmosphere.
• Research and development experience in:
• Model compression and neural architecture search techniques including knowledge distillation, pruning, quantization, and quantization-aware training; optimizing and deploying inference on diverse embedded processors.
• Expertise in defining compute architectures for efficient deep learning inferencing.
• Ability to comprehend hardware specifications and utilize performance profiling tools.
• Strong proficiency in Python programming and in-depth knowledge of at least one framework such as PyTorch, TensorFlow, or MXNet.
• Experience implementing inference logic from foundational principles using low-level routines like BLAS, CUDA Kernels, or native C++.
• Familiarity with ONNX runtimes and deep learning compilers like XLA or Glow.

Responsibilities:

- Design, refine, and implement ultra-low latency deep learning/machine learning algorithms for Rivians Advanced Driver-Assistance Systems (ADAS) and autonomous functionalities.

• Investigate cutting-edge model compression and efficient design strategies, enabling the team to apply these techniques across a variety of customer-focused features.
• Collaborate with low-level software and hardware architecture teams to assess the performance of our in-house ML models on embedded platforms and optimize them according to device computing and memory limitations.

Company:

At Rivian, we are driven by a mission to inspire adventurous living, both through our emissions-free Electric Adventure Vehicles and the bold individuals we seek to bring on board. We embrace a culture of innovation, always challenging the status quo and exploring new territories. Our team is diverse, united by a passion for the outdoors and a commitment to safeguarding it for future generations. As a Staff Software Engineer focused on ML acceleration and performance, you will join our Perception team, where you will contribute to critical machine learning algorithms that play a key role in the safety of our cutting-edge vehicles.

We offer a comprehensive benefits package, including robust medical, dental, and vision care, effective from your first day of employment.