Senior Deep Learning Frameworks Sustaining Engineer

NVIDIA is hiring software engineers for its Deep Learning Frameworks Sustaining Engineering team. Our team produces software that's powering a revolution in deep learning, enabling breakthroughs in problems from image classification to speech recognition to natural language processing! Help us build software which will be used by the entire world by working directly with deep learning framework developers by integrating with open source code bases.

What you'll be doing:

In this role, responsibilities will include back-porting changes from the mainline branch, keeping track of open source dependency changes, and ensuring the latest stable dependencies are being used for our enterprise products. You will actively contribute changes to the team to support timely Long Term Support releases for the TensorFlow, PyTorch, TensorRT products. The scope of these efforts includes fixing customer reported bugs, integrating bug fixes found in mainline and working with other teams to ensure open source dependencies are patched for security vulnerabilities to address the needs of NVIDIA AI Enterprise business subscribers.

What we need to see:

• Bachelors of Science in Computer Science, Deep Learning, Artificial Intelligence, Applied Math, or related field or equivalent experience.

• 5+ years of relevant software development experience

• Excellent C/C++ programming and software design skills, including debugging and open source integration. Python experience also helpful.

• Utilizing tools involved in building software (Make, Docker, Bazel), packaging systems (Debian, pip, npm, etc.), Build Systems (Gitlab, CI/Jenkins).

• Prior experience with machine learning algorithms and frameworks (TensorFlow, PyTorch, or MXNet).

• Ability to work independently, contribute to the stability of releases, and effectively communicate status to those involved in the release in a detail-oriented way.

Ways to stand out from the crowd:

• GPU programming experience (CUDA or OpenCL) desired but not required

• Experience with contributions to or managing large open source project - use of github, bug tracking, branching and merging code, OSS licensing issues, managing patches, etc.

• Familiarity with Gitlab CI pipelines

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people on the planet working for us. If you're creative and autonomous, we want to hear from you!

Your base salary will be determined based on your location, experience, and other objective job-related factors. The expected base salary ranges for this position are:
152,000 USD – 241,500 USD for Level 3; and 184,000 USD – 287,500 USD for Level 4. Applications for this job will be accepted at least until November 3, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive 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.