Compiler Optimization Engineer, MLIR and LLVM - New College Grad 2024

We are looking for a LLVM/MLIR Compiler Engineer for an exciting role in our Compute Compiler Team. We deliver features and improvements to CUDA and other compute compilers to better realize the potential of NVIDIA GPUs for a growing range of computational workloads, ranging from deep learning, scientific computation, and self-driving cars. Our compiler organization makes its mark on every GPU NVIDIA produces.

We need you as a key member of a small team that is working on a core compiler component for accelerating general purpose computation on the GPU. You will be solving critical problems working alongside some of the top valued diverse minds in GPU computing and systems software, doing what you enjoy. See your efforts in action as HPC and DL developers use features and optimizations to achieve the best performance of their applications. If this sounds like a fun challenge, we want to hear from you!

What you will be doing:

• Identify opportunities for performance improvements in the LLVM-based compiler middle end optimizer.
• Design and develop new compiler analysis passes and optimizations to produce best-in-class, robust, supportable compiler and tools.
• 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.
• Contribute to open-source projects (LLVM/MLIR).

What we need to see:

• Pursuing a M.S or Ph.D. in Computer Science, Computer Engineering, or related fields (or equivalent experience).
• Background 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.

Ways for you to stand out from the crowd:

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

NVIDIA is leading the way in groundbreaking developments in Artificial Intelligence, High-Performance Computing and Visualization. The GPU, our invention, serves as the visual cortex of modern computers and is at the heart of our products and services. Our work opens up new universes to explore, enables amazing creativity and discovery, and powers what were once science fiction inventions from artificial intelligence to autonomous cars. We are looking for great people like you to help us accelerate the next wave of artificial intelligence.

The base salary range is 104,000 USD - 189,750 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.