NVIDIA is looking for engineering and research interns to join its Deep Learning algorithms team. Academic and commercial groups around the world are using GPUs to revolutionize deep learning and data analytics, and to power data centers. Join the team building software which will be used by the entire world. Interact with the scientific community to implement the latest algorithms. Ability to work on a dynamic customer-oriented team is required and excellent interpersonal skills are also a requirement.
In this role you will be interacting with internal partners, users, and members of the open source community to analyze, define and implement highly optimized algorithms and DL frameworks. The scope of these efforts includes a combination of performance tuning and analysis, defining APIs, analyzing functionality coverage, implementing new algorithms and frameworks, and other general software engineering work.
What you’ll be doing:
Research, analyze, and document state-of-the art algorithms
Design and implement a deep learning framework for model optimization
Develop algorithms for deep learning, data analytics, machine learning, or scientific computing
Analyze performance of GPU implementations
Benchmark software stacks across training and inference scenarios
Evaluate and understand capabilities of frontier models
Collaborate with team members and other partners
What we need to see:
Pursuing MSc or PhD in Computer Science, Artificial Intelligence, Applied Math, or related field
Excellent programming in Python, debugging, performance analysis, and test design skills
Strong algorithms and mathematical fundamentals
Good understanding of Deep Learning fundamentals
Ability to work independently and manage your own development effort
Good communication and documentation habits
Ways to stand out from the crowd:
Deep Learning experience
Experience with DL Frameworks (PyTorch preferred) and Large Language Models
Experience with model compression techniques such as pruning, NAS, distillation, and quantization
Knowledge of CPU and/or GPU architecture
First-author publication in a top-tier deep learning or AI conference
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 working for us. If you're creative and autonomous, we want to hear from you!
We are an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.