Staff Software Engineer, ML Acceleration and Performance

Job Description

Rivian is looking for a Staff Software Engineer to work on ML Acceleration and Performance for their self-driving vehicles.

Requirements

• MS. or Ph.D. in Computer Science, Electrical, Mechanical, Aerospace Engineering or related field
• 7+ years of industrial experience in deep learning, heterogenous computing, ML accelerators, and compilers
• Strong Python programming background and knowledge of PyTorch, TensorFlow, or MXNet
• Experience with deploying inference on embedded processors and understanding hardware spec documents

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.