NVIDIA, Deep Learning Algorithm Engineering Intern

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

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