Deep Learning Compiler Intern

1 Month ago • Upto 1 Years • Artificial Intelligence

Job Summary

Job Description

As a Deep Learning Compiler Intern at NVIDIA, you'll contribute to the OpenAI Triton compiler, extending its support for new hardware, models, and programming paradigms. Responsibilities include analyzing deep learning networks, developing compiler optimization algorithms, collaborating with software and hardware teams, implementing compiler and optimization techniques, and performing performance optimization and general software engineering tasks. This internship offers a chance to work at the forefront of AI and accelerated computing with a top-notch team.
Must have:
  • C/C++ programming skills
  • Software design experience
  • Debugging, performance analysis
  • Test design skills
Good to have:
  • MLIR, LLVM, XLA, TVM experience
  • Deep learning models/algorithms knowledge
  • Deep learning framework design
  • Open-source contributions
  • CPU/GPU architecture knowledge
  • CUDA or OpenCL experience
Perks:
  • Intern benefits

Job Details

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 Intern 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.

Similar Jobs

Ubisoft - Senior R&D Engineer

Ubisoft

Pune, Maharashtra, India (On-Site)
2 Weeks ago
Nordcurrent - Unity Developer

Nordcurrent

Dnipro, Dnipropetrovsk Oblast, Ukraine (On-Site)
5 Months ago
Social Discovery Group - Head/Lead Social Media Manager

Social Discovery Group

Portugal (Remote)
6 Months ago
Google - Senior Software Engineer, Google Cloud Business Platforms

Google

Kirkland, Washington, United States (On-Site)
2 Weeks ago
Google - Group Product Manager, Google Cloud Storage

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago
Casumo - AI Engineer

Casumo

(Hybrid)
1 Month ago
Microsoft - Research Intern - Multimodal AI Research

Microsoft

Redmond, Washington, United States (On-Site)
2 Weeks ago
Hedra - Machine Learning Engineer (CUDA)

Hedra

New York, New York, United States (On-Site)
1 Month ago
Ubisoft - Senior C++ Programmer - Machine Learning

Ubisoft

Montreal, Quebec, Canada (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Inkittt - Marketing & Social Media Manager

Inkittt

San Francisco, California, United States (On-Site)
8 Months ago
PlayStation Global - Principal Software Engineer, Graphics

PlayStation Global

Aliso Viejo, California, United States (On-Site)
1 Month ago
Google - Software Engineer, Wi-Fi, Chrome OS

Google

Taipei City, Taiwan (On-Site)
2 Weeks ago
Google - Senior Hardware Qualification Engineer, Machine Learning, Google Cloud

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago
DraftKings - Full-Stack Engineer

DraftKings

Sofia, Sofia City Province, Bulgaria (Remote)
1 Month ago
Google - Software Engineer III, AI/ML GenAI, Google Ads

Google

Irvine, California, United States (On-Site)
1 Week ago
NVIDIA - Deep Learning Software Engineer, Performance Optimization

NVIDIA

Tokyo, Japan (On-Site)
3 Months ago
Blind Squirrel Games - Senior Graphics Engineer

Blind Squirrel Games

Irvine, California, United States (Hybrid)
6 Months ago
Genies - Machine Learning Infrastructure Engineer, 3D Model Inference & Deployment

Genies

San Mateo, California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in Santa Clara, California, United States

NVIDIA - Senior Site Reliability Engineer - GPU Clusters

NVIDIA

Austin, Texas, United States (On-Site)
2 Months ago
Salesforce - Tableau Account Executive, Federal Civilian

Salesforce

Virginia, United States (Remote)
1 Month ago
Google - Senior Digital Strategy Lead, Accelerated Growth Team

Google

New York, New York, United States (On-Site)
1 Week ago
Evolution - Live In-Studio Game Presenter

Evolution

Atlantic City, New Jersey, United States (On-Site)
2 Months ago
The Walt Disney Company - Principal Data Engineer

The Walt Disney Company

Seattle, Washington, United States (On-Site)
2 Months ago
ByteDance - Research Scientist, Multimodality

ByteDance

San Jose, California, United States (On-Site)
6 Months ago
Samsung Semiconductor - Staff Engineer, Formal Verification

Samsung Semiconductor

San Jose, California, United States (Hybrid)
1 Month ago
Fluence - Senior Product Compliance Engineer

Fluence

Houston, Texas, United States (Hybrid)
6 Months ago
Daybreak Game Company LLC - Director / Senior Manager, People & Culture

Daybreak Game Company LLC

San Diego, California, United States (Remote)
2 Weeks ago
ByteDance - Network Software Development Engineer, High Speed Network

ByteDance

Seattle, Washington, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Scale AI - Software Engineer - New Grad

Scale AI

San Francisco, California, United States (On-Site)
6 Months ago
Enterprise Bot - Data Scientist

Enterprise Bot

Bengaluru, Karnataka, India (On-Site)
5 Months ago
Krafton  - Strategy Manager (AI Ethics)

Krafton

Seoul, South Korea (On-Site)
1 Month ago
Meta - Software Engineer, Machine Learning

Meta

Burlingame, California, United States (On-Site)
5 Months ago
Rackspace Technology - Principal MLOps Engineer

Rackspace Technology

(Remote)
1 Month ago
NVIDIA - Senior Developer Technology Engineer - AI

NVIDIA

Westford, Massachusetts, United States (Hybrid)
1 Month ago
Ubisoft - Senior ML Data Scientist

Ubisoft

Montreal, Quebec, Canada (On-Site)
1 Month ago
Google - Senior AI Sales Specialist, Google Cloud

Google

Tokyo, Japan (On-Site)
1 Week ago
Google - Software Engineer III, AI/ML GenAI, Google Cloud Data Management

Google

Sunnyvale, California, United States (On-Site)
2 Weeks ago
N-iX - Senior AI/GenAI Solution Engineer

N-iX

(Remote)
2 Weeks ago

Get notifed when new similar jobs are uploaded

About The Company

Since its founding in 1993, NVIDIA (NASDAQ: NVDA) has been a pioneer in accelerated computing. The company’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined computer graphics, ignited the era of modern AI and is fueling the creation of the metaverse. NVIDIA is now a full-stack computing company with data-center-scale offerings that are reshaping industry.

Santa Clara, California, United States (On-Site)

Santa Clara, California, United States (On-Site)

Massachusetts, United States (On-Site)

Santa Clara, California, United States (On-Site)

Santa Clara, California, United States (On-Site)

Santa Clara, California, United States (On-Site)

Texas, United States (On-Site)

Santa Clara, California, United States (Hybrid)

Austin, Texas, United States (Remote)

Santa Clara, California, United States (Hybrid)

View All Jobs

Get notified when new jobs are added by NVIDIA

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug