Senior Ai Compiler Engineer, Mlir

NVIDIA

Base: $152,000 - $241,500 usd; bonus/equity: eligi...
Not specified (potentially hybrid or fully remote based on company norms)
Mlir compiler technologies
Llvm and xla experience
C/c++ and python programming
NVIDIA is seeking a Senior AI Compiler Engineer to develop MLIR-based AI compilers for their inference engine, focusing on performance and usability. The ideal candidate should have a strong background in compiler technologies and relevant programming skills, with experience in machine learning models being a plus

Job Summary

  • This role involves building an MLIR-based AI compiler that powers NVIDIA's inference engine end-to-end for data center and edge applications.
  • Candidates must possess strong C/C++ and Python skills to define APIs, implement optimizations, and generate kernels for neural networks.
  • The position offers competitive salaries ranging from $152,000 to $241,500 USD along with equity and generous benefits.

Matching Summary

Match Score: 85

NVIDIA is seeking a Senior AI Compiler Engineer to develop MLIR-based AI compilers for their inference engine, focusing on performance and usability. The ideal candidate should have a strong background in compiler technologies and relevant programming skills, with experience in machine learning models being a plus.

Salary

Base: $152,000 - $241,500 USD; Bonus/Equity: Eligible for equity; Benefits: Generous benefits package

Skills & Requirements

Must-have

  • MLIR compiler technologies
  • LLVM and XLA experience
  • C/C++ and Python programming
  • Performance analysis skills
  • Compiler optimization expertise

Nice-to-have

  • Deep learning model understanding
  • PyTorch and JAX frameworks
  • GPU kernel generation experience
  • CUDA or OpenCL proficiency
  • Mentoring early career engineers

Key Requirements

  • Bachelor's, Master's, or Ph.D. in Computer Science
  • 3+ years of relevant work or research experience
  • Experience with compiler technologies such as MLIR, XLA, and LLVM

Work Rights

Not specified

Tailored Resume

Cover Letter