Staff Machine Learning Compiler Engineer

$206,500 - $258,100/Yr

Rivian - Palo Alto, CA

Full-time - Mid Level

Palo Alto, CA

Transportation Equipment Manufacturing

About the position

The Staff Machine Learning Compiler Engineer will be a vital member of the ML Compiler team at Rivian, focusing on developing software tools that enable the inference of deep learning networks on Rivian's hardware platforms. This role involves collaboration with the ADAS and Hardware teams to evaluate performance and implement new hardware features in the compiler.

Responsibilities

• Develop fast and accurate simulation models for Rivian Hardware Platform
• Conduct performance analysis and optimization of ML workloads
• Collaborate with ADAS and software teams to facilitate the migration of software models to hardware

Requirements

• Degree in Computer Engineering or a related field
• Excellent C/C++ and Python programming skills
• Experience with various SOC platforms used for machine learning and ADAS
• Strong understanding of deep learning software models
• Experience in compiler pipeline development preferred
• Experience working in aggressive design environments preferred

Benefits

• Dental insurance
• Vision insurance