Senior Deep Learning Frameworks Sustaining Engineer

Nvidia
US, CA, Santa Clara2026-01-09onsite

About the job

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.

Responsibilities

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.

Qualifications

Minimum

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.

Preferred

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