Compiler Engineer - New Grad

Job summary

Responsibilities

As a new grad Compiler Engineer, you'll work on core areas of our compiler, including the IR and pass infrastructure, model lowering and partitioning, tensor tiling, memory management, and mapping operations onto the fabric.

• Implement and optimize passes in the compiler's IR and pass infrastructure
• Lower and partition machine learning models for execution on the RDU
• Develop tensor tiling and memory management for on-chip memory
• Map operations onto the RDU fabric to improve model performance
• Debug compiler and performance issues at a low level in C++ and Python

What they're looking for

• B.S. in Computer Science, Electrical Engineering, Computer Engineering, or related field
• Experience in machine learning systems development, including ML compilers, LLM inference kernels, or kernel generation
• Strong programming skills in Python and C/C++
• Experience programming GPUs or other AI accelerators
• M.S. or Ph.D. in Computer Science, Electrical Engineering, Computer Engineering, or related field

Tech stack

• C++
• Python