Are you passionate about compiler technology and computer architectures for deep learning? Do you thrive at the intersection of hardware and software? NVIDIA is seeking world-class compiler engineers and performance architects who are excited to push the boundaries of machine learning infrastructure. In this role, you will develop and optimize MLIR-based compiler infrastructure that powers our deep learning libraries and influences the direction of future GPU architectures. This position offers the opportunity to make a significant impact in a fast-moving, technology-focused company.

What You'll Be Doing:

• Design, implement, and optimize MLIR-based compiler passes for deep learning and data analytics workloads.

• Analyze and improve the performance of machine learning and deep learning algorithms on current and next-generation architectures with compiler technologies.

• Identify performance bottlenecks in compiler-generated code and propose creative solutions.

• Collaborate with hardware architects and software teams to co-design features that maximize performance and efficiency.

• Contribute to the evolution of NVIDIA’s deep learning compiler stack and libraries.

What We Need to See:

• MS or PhD in Computer Science, Electrical Engineering, Mathematics, or a related field, or equivalent experience.

• 5+ years of working experience

• Proven experience developing compilers or compiler infrastructure, preferably with MLIR, LLVM, or similar frameworks.

• Strong programming skills in C++ and Python.

• Solid understanding of computer architecture, especially as it relates to performance optimization.

• Experience optimizing code for CPUs or GPUs, including low-level programming (assembly, SIMD, or vectorization).

• Experience with deep learning algorithms, especially matrix multiplication and convolution.

Ways to Stand Out from the Crowd:

• Hands-on experience with MLIR, LLVM, or other modern compiler frameworks.

• Deep understanding of parallel programming models and GPU architectures.

• Strong communication and organizational skills.

• Demonstrated ability to work collaboratively in a fast-paced, cross-functional environment.

If you are excited about building the next generation of machine learning compilers and want to work with world-class teams at the forefront of AI and hardware innovation, we want to hear from you!