Are you passionate about blazing the unbounded field of deep learning and building the best software tools for developers? We are looking for an outstanding software engineer to join the Developer Tools organization at NVIDIA that aims at delivering the best-in-class tool to enable streamlined design and optimization of high-performance deep neural network models in a unified environment. Come join the team and see how you can make a lasting impact on the world!

What You Will Be Doing:

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

What We Need to See:

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

Ways to Stand Out from the Crowd:

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

NVIDIA is widely regarded as one of the most innovative technology companies in the industry, and we have some of the most forward-thinking and versatile people in the world. Are you creative and autonomous? Do you love challenges and make a difference? If so, we want to hear from you.