Compiler Engineer, New Grad

About the Role

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. Compiler and hardware design are closely linked here, and you'll see how compiler decisions affect chip performance. You'll write production code in C++ and Python, debug at a low level, and work with the runtime and hardware teams. Senior engineers will review your code and mentor you as you take on larger parts of the compiler.

Key Responsibilities

• 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

Requirements & Tech Stack

• 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

Qualifications

• M.S. or Ph.D. in Computer Science, Electrical Engineering, Computer Engineering, or related field
• Experience with ML compiler frameworks such as MLIR or StableHLO
• Experience with LLM kernel libraries
• Machine learning systems research experience
• Open source project ownership or contributions
• Experience building AI agent driven tools or automation