Senior Compiler Optimization Engineer - MLIR

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.
• Strong background in software engineering and compiler optimization.
• Ability to work effectively in a collaborative, interdisciplinary team environment.

Benefits

• Highest competitive salaries.
• Comprehensive benefits package.
• Equity options.

Pay

Base salary range: $184,000 - $287,500 USD.

Schedule

Full-time position.