Senior AI Compiler Engineer, MLIR

Nvidia

Confirmed live 2 days ago Trusted
Remote

Quick summary

Work type
Remote
Location
Santa Clara, CAAustin, TX
Salary
$152,000–$241,500 / yr
Posted
28 days ago
Freshness
Confirmed live 2 days ago

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $211k
This role $197k
$140k most similar roles pay here $265k

This role pays less than 59% of similar roles. Most pay $176,000–$246,150 — the shaded band above. At the midpoint, this role pays about $197k versus about $211k for comparable roles.

Based on 240 similar postings.

Employer

About Nvidia

Nvidia is a leading designer of graphics processing units (GPUs) and system-on-chip units, powering gaming, professional visualization, data centers, and artificial intelligence workloads. Industry: Semiconductors & AI Computing

Nvidia currently has 896 open roles on FindRole.

Listed pay typically runs $184,000–$287,500 across 876 roles with salary data.

Most-posted roles

View all roles at Nvidia

At a glance

TL;DR · Senior AI Compiler Engineer, MLIR

As a Senior AI Compiler Engineer, MLIR, you will join the team responsible for building an MLIR-based AI compiler that powers the inference engine end to end. Your daily responsibilities include developing MLIR-based graph representations and optimizations for future GPU architectures while partnering with framework and hardware teams to enable new model patterns and architectural features. You will define APIs and MLIR dialects, conduct performance analysis, implement compiler optimizations, and perform kernel generation for neural networks. The role requires proficiency in C++, Python, MLIR, XLA, and LLVM. Candidates should possess experience in performance analysis, compiler optimizations, and GPU kernel generation. This position focuses on the technical challenge of optimizing inference across data center and edge environments by improving performance, build speeds, and memory usage for deep learning models like those found in generative AI and vision systems.

What you'll do

  • Develop MLIR-based graph representations and optimizations for future GPU architectures.
  • Define APIs and MLIR dialects to support inference engine capabilities.
  • Implement compiler optimizations and kernel generation specifically for neural networks.
  • Conduct performance analysis and optimization to ensure fast builds and low memory usage.
  • Partner with framework and hardware teams to enable new model patterns and architectural features.
  • Develop software for both Ahead-of-Time and Just-in-Time usability across data center and edge environments.

What we're looking for

  • Bachelor's, Master's, or Ph.D. in Computer Science, Computer Engineering, a related field, 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 development efforts.
  • Understanding of deep learning models, algorithms, and frameworks such as PyTorch and JAX (preferred).
  • Experience with GPU kernel generation targeting high performance and fast build times (preferred).
  • Proficiency in GPU architecture with CUDA or OpenCL programming experience (preferred).

More like this

Similar roles

Senior Deep Learning Compiler Engineer

Nvidia

Remote (Santa Clara, CA) +1 134 days ago $152,000$241,500
C++ Python CUDA OpenCL MLIR XLA TVM LLVM PyTorch GPU Architecture Compiler Optimization Deep Learning Kernel Generation cross compilation Performance Analysis
3+ yrs exp Remote

Senior Deep Learning Compiler Engineer, XLA

Nvidia

Remote 28 days ago $152,000$241,500
C++ CUDA XLA MLIR LLVM OpenAI Triton JAX PyTorch TensorFlow TVM High-Performance Computing Distributed Programming Compiler Optimization Deep Learning
4+ yrs exp Remote

Senior DL Compiler Engineer, CUDA Tile

Nvidia

Remote (Santa Clara, CA) +4 9 days ago $152,000$241,500
CUDA C++ C MLIR LLVM XLA TVM Compiler Optimization GPU Architecture Performance Analysis Deep Learning IR Design OpenCL
3+ yrs exp Remote

Deep Learning Compiler Engineer

Nvidia

Remote (Santa Clara, CA) +1 19 days ago $152,000$241,500
CUDA C++ C MLIR LLVM XLA TVM Compiler Optimization GPU Architecture Performance Analysis Deep Learning IR Design OpenCL
3+ yrs exp Remote

Senior Compiler Engineer, AI

Nvidia

Remote (Austin, TX) +2 119 days ago $184,000$287,500
LLVM MLIR Python C++ Machine Learning Reinforcement Learning Compiler Design GPU Architecture Code Generation Predictive Modeling Evolutionary Algorithms Feature Engineering High-Dimensional Data Analysis Performance Engineering
8+ yrs exp Remote