Senior Compiler Optimization Engineer - MLIR

at Nvidia

📍 Redmond, United States

USD 184,000-287,500 per year

SENIOR

✅ On-site

Tech Stack
Tag name is followed by "@" symbol and proficiency level value. About proficiency levels:

1-2 — basic awareness. Minimal hands-on experience, and a rudimentary understanding of the technology's purpose;

3-6 — daily use. Comfortable and regular usage, capable of handling common tasks and challenges related to the technology;

7-9 — you are an expert, you can teach others, you know all the pitfalls and tricks;

10 — exceptional knowledge, comprehensive understanding, and adeptness in all aspects of the technology, including advanced problem-solving. Think twice before claiming or demanding such level.

AI CUDA @ 4 Communication @ 6 Deep Learning @ 3 GPU @ 4 LLVM @ 4 Parallel Programming @ 4 Profiling

NVIDIA is seeking an experienced Compiler Optimization Engineer for an exciting role in its Compute Compiler Team.

Analyze the performance of application code running on NVIDIA GPUs with the aid of profiling tools.
Identify opportunities for performance improvements in the MLIR/LLVM based compiler middle end optimizer.
Identify and implement novel ideas in the compilation pipeline to enable overall best-in-class performance of AI workloads.
Design and develop new compiler passes and optimizations to produce best-in-class, robust, supportable compiler and tools.
Interact with the Open-source LLVM community to ensure tighter integration.
Work with geographically distributed compiler, hardware and application teams to oversee improvements and problem resolutions.
Be part of a team centered on deep-learning compiler technology spanning architecture design and support through higher level languages.

B.S., M.S. or Ph.D. in Computer Science, Computer Engineering, or related fields (or equivalent experience).
8+ years experience in Compiler Optimizations such as Loop Optimizations, Inter-procedural optimizations and Global optimizations.
Excellent hands-on C++ programming skills.
Understanding of any Processor ISA (GPU ISA would be a plus).
Strong background in software engineering principles with a focus on crafting robust and maintainable solutions.
Good communication and documentation skills and self-motivated.

Masters or PhD preferred.
Experience developing applications in CUDA or other parallel programming language.
Deep understanding of parallel programming concepts.
MLIR, LLVM and/or Clang compiler development experience.
Familiarity with deep learning frameworks and NVIDIA GPUs.

NVIDIA base salary range: 184,000 USD - 287,500 USD. You may also be eligible for equity and benefits.