Senior Compiler Optimization Engineer - MLIR
NVIDIA · Santa Clara, CA · 2 wk ago
EngineeringFull-time
About the role
We are looking for an experienced Compiler Optimization Engineer for an exciting role in our Compute Compiler Team. Our team delivers features and improvements to CUDA and other compute compilers to better realize the potential of NVIDIA GPUs for a growing range of computational workloads, including deep learning, scientific computation, and self-driving cars.
Responsibilities
- 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 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 that is at the center of deep-learning compiler technology spanning architecture design and support through higher-level languages.
Requirements
- 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 to challenging problems.
- Good communication and documentation skills and self-motivated.
Qualifications
- Masters or PhD preferred.
- Experience in developing applications in CUDA or other parallel programming languages.
- Deep understanding of parallel programming concepts.
- MLIR, LLVM, and/or Clang compiler development experience.
- Familiarity with deep learning frameworks and NVIDIA GPUs.
Skills
- Experience with NVIDIA GPUs and deep learning frameworks.
- Knowledge of compiler optimization techniques and LLVM/MLIR.
- Ability to work effectively in a cross-functional team setting.
- Strong analytical and problem-solving skills.
Benefits
We offer highly competitive salaries and a comprehensive benefits package, including equity and benefits. Applications for this job will be accepted at least until July 24, 2026. This posting is for an existing vacancy. NVIDIA is committed to fostering an inclusive work environment and is an equal opportunity employer.