Senior AI Compiler Engineer, MLIR

NVIDIA
Santa Clara, Austin, Seattle
Workplace: OnsiteFull timeUSD 152,000 - 287,500 annuallyFunction: Data Science & Machine LearningExperience: 3+ yearsSkills: ["Interpersonal skills","Independent work"]

Build an MLIR-based AI compiler that powers NVIDIA’s inference engine end to end, with an emphasis on performance, fast builds, and low memory use. Develop MLIR graph representations and optimizations for future GPU architectures, define MLIR dialects and APIs, and implement compiler optimizations and kernel generation for neural networks. Partner with framework and hardware teams to enable new model patterns and GPU architectural features across data center and edge.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
NVIDIA
NVIDIA
4 days ago

Senior AI Compiler Engineer, MLIR

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 3 hours agoStatus: Live
Reposted: similar role first listed 5 months ago

Job Summary

Build an MLIR-based AI compiler that powers NVIDIA’s inference engine end to end, with an emphasis on performance, fast builds, and low memory use. Develop MLIR graph representations and optimizations for future GPU architectures, define MLIR dialects and APIs, and implement compiler optimizations and kernel generation for neural networks. Partner with framework and hardware teams to enable new model patterns and GPU architectural features across data center and edge.
Location: Santa Clara, Austin, Seattle
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •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, then conduct performance optimizations and analysis.
  • •Implement compiler optimizations and kernel generation for neural networks.
  • •Contribute to general software engineering work as needed.

Pay and Benefits

Salary: USD 152,000 - 287,500 annually
Equity and Bonus:Equity

Key Requirements

  • •Bachelor’s, Master’s, or Ph.D. in Computer Science, Computer Engineering, or a related field, or equivalent experience.
  • •3+ years of relevant work or research experience in performance analysis and compiler optimizations.
  • •Experience with compiler technologies including MLIR, XLA, and LLVM.
  • •Strong C/C++ and Python programming skills plus software design, debugging, performance analysis, and testing.
  • •Ability to work independently, define project goals and scope, and lead your own development efforts.
Experience:3+ years
Education:
Skills:Interpersonal skillsIndependent work
Tech Stack:MLIRXLALLVMCC++PythonPyTorchJAXCUDAOpenCL

Company Brief

NVIDIA
Designs and manufactures GPUs, AI accelerators, and system-on-chip products for gaming, data centers, professional visualization, and automotive markets, enabling advanced graphics, AI, and high-performance computing solutions worldwide.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1993
Glassdoor
Glassdoor: 4.3
WebsiteLinkedInGlassdoor