Senior Deep Learning Software Engineer – Developer Tools Full-time

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 seeks to deliver the best-in-class tool to enable streamlined design and optimization of high-performance deep neural network models in a unified environment. Come be part of our team!

What You Will Be Doing:

• Innovate on tools support for design and performance optimizations of deep neural network models.
• Partner with management and architects to translate requirements into designs and lead development.
• Stay up to date with the latest research and become a domain authority 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 knowledge of deep learning and hands-on experience working with 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 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 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 a challenge and make a difference? If so, we want to hear from you.

#LI-Hybrid

The base salary range is 148,000 USD – 276,000 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

You will also be eligible for equity and benefits. NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.