# Senior AI Compiler Engineer, MLIR

**Company**: NVIDIA
**Experience**: senior
**Job type**: full-time
**Category**: Engineering
**Industry**: Technology

**Apply**: https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-AI-Compiler-Engineer--MLIR_JR2022727?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_5b928975-d67

## Description

NVIDIA is hiring a Senior AI Compiler Engineer. You will build an MLIR-based AI compiler that powers NVIDIA's inference engine end to end, with a focus on performance, fast builds, low memory use, and Ahead-of-Time and Just-in-Time usability across data center and edge.

**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, conduct performance optimizations and analysis, implement compiler optimizations and kernel generation for neural networks, and contribute to other general software engineering work.

**Requirements:**

- 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 your own development efforts.

- Strong interpersonal skills and the ability to thrive in a fast-moving, dynamic, product-oriented team.

**Preferred Qualifications:**

- Understanding of deep learning models, algorithms, and frameworks such as PyTorch and JAX.

- Experience with GPU kernel generation targeting high performance and fast build times.

- Proficiency in GPU architecture with CUDA or OpenCL programming experience.

- A track record of mentoring early career engineers and interns is a bonus.

With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most desirable employers.

## Skills

### Required
- MLIR
- XLA
- LLVM
- C/C++
- Python
- compiler technologies
- performance analysis
- compiler optimizations

### Nice to have
- deep learning models
- PyTorch
- JAX
- GPU kernel generation
- CUDA
- OpenCL
- GPU architecture

---

Source: [Apply at nvidia.wd5.myworkdayjobs.com](https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/US-CA-Santa-Clara/Senior-AI-Compiler-Engineer--MLIR_JR2022727?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)