AI Compiler Engineer (New Grad 2027) – Equity Eligible

NVIDIA AI; Santa Clara (CA); On-site; USD 108,000 - 178,000; Full time

Job summary

NVIDIA in Santa Clara, CA, is hiring outstanding AI compiler engineers to develop groundbreaking technologies in machine learning compilers and AI systems. You will create compiler abstractions and accelerate ML workloads, partnering with NVIDIA software and hardware teams to push latest developments into products.

The role demands a strong background in ML systems, ML compilers, and GPU programming, with experience in Python and C/C++.

Qualifications

• Bachelor's degree in CS/EE or related field; MS/PhD preferred.
• Experience in ML systems development including ML compilers and LLM inference kernels.
• Strong Python, C/C++, and GPU programming skills.

Responsibilities

• Innovate and develop ML compiler and systems technologies.
• Design, implement, and optimize compilers for high impact AI workloads.
• Build agent driven solutions to automate the overall development flow.
• Co-design agent system solutions with current and future AI compiler technologies.
• Collaborate closely with NVIDIA engineering teams to build high impact solutions for ML acceleration.

Skills

Python

C/C++

GPU programming

Education

Bachelor's degree in Computer Science, Electrical Engineering, or related field

MS or PhD preferred

Tools

Apache TVM

MLIR

FlashInfer

Flash Attention

Job description

NVIDIA in Santa Clara, CA, is hiring outstanding AI compiler engineers to develop groundbreaking technologies in machine learning compilers and AI systems. You will create compiler abstractions and accelerate ML workloads, partnering with NVIDIA software and hardware teams to push latest developments into products.

The role demands a strong background in ML systems, ML compilers, and GPU programming, with experience in Python and C/C++.