Deep Learning Compiler Intern

NVIDIA is widely recognized as a pioneer in visual computing and artificial intelligence. Our AI Compiler team is at the cutting edge of technology, pushing the boundaries of what GPUs can achieve. As an intern with our team, you'll be working on the OpenAI Triton compiler, an innovative project that aims to make high performance kernel development more accessible, and you'll be helping to extend it to support new hardware, models, and programming paradigms. This role offers a unique opportunity to collaborate with some of the brightest minds in the industry, contributing at the frontier intersection of AI and accelerated computing.

We are now hiring a Deep Learning Compiler Intern!

Our GPUs are at the center of the deep learning revolution and continue to enable breakthroughs in generative AI, large language models, recommendation systems, speech recognition, image classification and other areas. Come join us to work with a top-notch team and have broad impact across the entire deep learning community.

What you'll be doing:

• Analyze deep learning networks and developing compiler optimization algorithms.
• Collaborate with members of the deep learning software framework teams and the hardware architecture teams to accelerate the next generation of deep learning software.
• The scope includes crafting and implementing compiler and optimization techniques, performance optimization, and other general software engineering work.

What we need to see:

• Currently pursuing a Bachelor's, Master's, or PhD degree within Computer Engineering, Electrical Engineering, Computer Science, or a related field
• Excellent C/C++ programming and software design skills
• Experience with debugging, performance analysis, and test design.

Ways to stand out from the crowd:

• Experience with the following technologies: MLIR, LLVM, XLA, TVM, deep learning models and algorithms, and deep learning framework design.
• History of contributions to open-source projects.
• Knowledge of CPU and/or GPU architecture. CUDA or OpenCL programming experience.

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world on our team. If you are creative and autonomous, we want to hear from you!

The hourly rate for our interns is 18 USD - 71 USD. Our internship hourly rates are a standard pay determined based on the position and your location, year in school, degree, and experience.

You will also be eligible for Internbenefits. 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.