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 tool support for deep neural network design and performance, translating requirements into designs, and staying updated with the latest research in deep learning.

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