Senior AI Compiler Engineer, MLIR

NVIDIA

Senior AI Compiler Engineer building MLIR-based AI compilers to optimize GPU inference at NVIDIA. Collaborating with hardware teams, focusing on performance and usability across data centers and edge devices.

Posted 4/22/2026full-timeRemote • California, Texas, Washington • 🇺🇸 United StatesSenior💰 $152,000 - $241,500 per yearWebsite

Tech Stack

Tools & technologies

Python

About the role

Key responsibilities & impact

Develop MLIR-based graph representations and optimizations for future GPU architectures
Partner with framework and hardware teams to enable new model patterns and upcoming GPU architectural features
Define APIs and MLIR dialects, conduct performance optimizations and analysis, implement compiler optimizations and kernel generation for neural networks
Contribute to other general software engineering work

Requirements

What you’ll need

Bachelor's, Master's, or Ph.D. in Computer Science, Computer Engineering, or equivalent experience
3+ years of relevant work or research experience in performance analysis and compiler optimizations
Experience with compiler technologies such as MLIR, XLA, and LLVM
Excellent C/C++ and Python programming and software design skills, including debugging, performance analysis, and testing
Ability to work independently, define project goals and scope, and lead your own development efforts
Strong interpersonal skills and the ability to thrive in a fast-moving, dynamic, product-oriented team

Benefits

Comp & perks

competitive salaries
generous benefits package
eligible for equity

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

MLIRXLALLVMCC++Pythonperformance analysiscompiler optimizationskernel generationsoftware design

Soft Skills

independent workproject goal definitionleadershipinterpersonal skillsadaptabilityteam collaborationcommunication

Certifications

Bachelor's degreeMaster's degreePh.D.