UIUC Research Park Intern - Machine Learning… at Rivian · Champaign

Summary: Rivian is on a mission to keep the world adventurous forever, and they are seeking an intern for the ML Compiler team. This role involves working on software tools to enable inference of deep learning networks on Rivian hardware, collaborating closely with the Autonomy and Hardware teams.

Responsibilities:
- Work on the ML Compiler for mapping Autonomy ML models to Rivian Autonomy Processor (RAP1)
- Implement graph lowering passes in the compiler to enable compiling various models to RAP1
- Implement hardware-aware optimizations, memory-efficient representations, and operator fusion, targeted to RAP1
- Benchmark and analyze system performance across platforms and iterate to achieve optimal deployment efficiency
- Verify functionality through functional simulation or emulation

Required Qualifications:
- Must be currently pursuing a bachelors, masters, or PhD degree at the University of Illinois Urbana Champaign
- Actively pursuing a degree or one closely related in Engineering
- Good understanding of algorithms for Computer Vision applications, Deep Learning, Natural Language Processing, Artificial Intelligence or related fields
- Experience with ML compilation stacks
- Strong mathematical background and capability to analyze results
- Good understanding of software fundamentals
- Excellent C++ and Python programming skills
- Must be enrolled in the University of Illinois Urbana Champaign
- Must be able to work out of our office in the Research Park
- Must be available for 20 hours per week in the Fall and Spring and 40 hours in the summer
- This project will require weekly task status updates and monthly project updates

Required Skills: Machine Learning Compiler, Deep Learning, Computer Vision, Natural Language Processing, Artificial Intelligence, ML Compilation Stacks, Mathematics, C++, Python

Benefits: Paid vacation, Paid sick leave, Medical insurance benefits, Internship positions are not eligible for retirement benefits.