Staff Software Engineer, ML Acceleration and Performance

Job Description

Rivian is seeking a Staff Software Engineer for ML Acceleration and Performance to work on advanced machine learning algorithms for self-driving features of their Electric Adventure Vehicles.

Requirements

• MS. or Ph.D. in Computer Science, Electrical, Mechanical, Aerospace Engineering or a related field
• 7+ years of industrial experience in deep learning, heterogenous computing, ML accelerators, and compilers
• Experience in model compression, neural architecture search, knowledge distillation, pruning, quantization, and deploying inference on embedded processors
• Strong Python programming background and familiarity with PyTorch, TensorFlow, or MXNet
• Experience with low-level subroutines like BLAS, CUDA Kernels, or C++ for implementing inference logic
• Experience with hardware spec documents and performance profiling tools

Responsibilities

• Develop, optimize, and deploy ultra-low latency deep learning/machine learning algorithms for Rivian ADAS and Autonomy use cases
• Research state-of-the-art model compression and efficient model design techniques
• Collaborate with low-level software and hardware architecture teams to optimize ML models for embedded platforms

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.