Deep Learning Performance Architect Intern null

About NVIDIA

NVIDIA pioneered accelerated computing to tackle challenges no one else can solve. NVIDIA's work in AI and digital twins is transforming the world's largest industries and profoundly impacting society.

Your role

NVIDIA is looking for world-class programmers and performance architects who like to continuously explore and mine the ultimate performance of each operator (or fusion operator) in deep learning networks, design and develop scalable modular infrastructure that can ship these highly optimized operators to different NVIDIA software libraries for training and inference.

What you'll be doing:

• Analyze the performance of various machine learning/DL algorithms on existing/new architectures
• Identify bottlenecks and propose creative solutions to improve them
• Develop high-performance operators on NVIDIA GPUs for cuBLAS, TensorRT, cuDNN, cuSparse, and cuTensor libraries
• Design and develop software for shipping and testing the GPU operators
• Build scalable automation for testing, integration, and release processes for publicly distributed deep learning libraries
• Configure, maintain and build upon deployments of industry-standard tools (e.g., Kubernetes, Jenkins, Docker, CMake, Gitlab, Jira, etc)

Training & development

The NVIDIA Deep Learning Institute (DLI) offers resources for diverse learning needs—from learning materials to self-paced and live training to educator programs. Individuals, teams, organizations, educators, and students can now find everything they need to advance their knowledge in AI, accelerated computing, accelerated data science, graphics and simulation, and more.

Benefits

NVIDIA offers interns the following:

• Intern Experience
• Healthcare Programs
• Time Off & Holidays
• Well-Being
• Learning & Development

Career progression

At NVIDIA, the mission is to challenge, inspire, and offer amazing opportunities for the continuous growth and learning of its interns. NDIVIA makes it happen through a comprehensive benefits package that includes engaging programs, exciting events, special perks, and more.

Work-life balance

NVIDIA is flexible and supports the most effective way for you to work, whether it’s from home, in the office, or a combination of the two. You’re also eligible for all holidays and free Days. Use these days to disconnect and recharge.

Culture & vibe

NVIDIA's culture is centered around innovation, collaboration, and a drive for excellence. The company fosters an environment where employees are encouraged to push the boundaries of technology and creativity. NVIDIA emphasizes agility, inclusiveness, and a commitment to learning and growth, making it a place where employees can truly do their life's work.

About you

What NVIDIA needs to see:

• Pursuing a B.S., M.S., or PhD degree in computer science (or similar)
• Strong programming skills in C/C++ development
• Familiar with GPU programming model and CUDA
• Good understanding of AI compilation technologies and experience with MLIR, TVM development
• Excellent problem-solving skills, good communication, and teamwork

How to apply

NVIDIA hiring process:

• Search and apply
• Meet
• Join the team

Sources

The following sources were used in researching this page:

• nvidia.com/en-us/benefits/interns/
• nvidia.com/en-us/about-nvidia/careers/university-recruiting/
• learn.nvidia.com/
• nvidia.com/en-in/about-nvidia/culture-at-nvidia/