Senior Software Engineer - Developer Tools for Deep Learning

NVIDIA is widely regarded as one of the most innovative technology companies in the industry, and they are seeking a Senior Software Engineer to join their Developer Tools organization. The role focuses on enhancing software tools for deep learning to optimize high-performance deep neural network models and requires collaboration with management and architects to translate requirements into designs.

Responsibilities:

• Enhance tool support to improve deep neural network design and performance efficiency
• Partner with management and architects to translate requirements into designs and own the development
• Stay up to date with the latest research and become a domain expert in deep learning

Requirements:

• Master's degree in computer science, mathematics, or related engineering fields (or equivalent experience)
• At least 5 years of relevant technical work experience
• Excellent foundational knowledge of deep learning and hands-on experience working with SOTA computer vision and large language models
• Working experience with DL frameworks such as PyTorch, Tensorflow, or JAX
• Strong programming skills in Python or C++
• Excellent communication and interpersonal skills
• Self-motivated and ability to work effectively with geographically distributed and multi-functional teams
• In-depth knowledge of PyTorch and ONNX
• Familiar with NVIDIA DL software stacks (TensorRT, etc.)
• Hands-on experience with developing real-world deep learning applications, from training to deployment
• Experience and/or interests in building software tools for designing and optimizing deep neural network models