Senior Deep Learning Frameworks Sustaining Engineer jobs in United States
info-icon
This job has closed.
company-logo

NVIDIA · 2 days ago

Senior Deep Learning Frameworks Sustaining Engineer

NVIDIA is hiring software engineers for its Deep Learning Frameworks Sustaining Engineering team. The role involves back-porting changes, managing open source dependencies, and ensuring stable dependencies for enterprise products while contributing to timely Long Term Support releases for TensorFlow, PyTorch, and TensorRT.

AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

Back-porting changes from the mainline branch
Keeping track of open source dependency changes
Ensuring the latest stable dependencies are being used for our enterprise products
Actively contribute changes to the team to support timely Long Term Support releases for the TensorFlow, PyTorch, TensorRT products
Fixing customer reported bugs
Integrating bug fixes found in mainline
Working with other teams to ensure open source dependencies are patched for security vulnerabilities to address the needs of NVIDIA AI Enterprise business subscribers

Qualification

C/C++ programmingDeep Learning frameworksSoftware design skillsPythonMachine learning algorithmsBuild systemsGPU programmingOpen source integrationGitlab CI pipelines

Required

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

Benefits

Equity
Benefits

Company

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.

H1B Sponsorship

NVIDIA has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1877)
2024 (1355)
2023 (976)
2022 (835)
2021 (601)
2020 (529)

Funding

Current Stage
Public Company
Total Funding
$4.09B
Key Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity

Leadership Team

leader-logo
Jensen Huang
Founder and CEO
linkedin
leader-logo
Michael Kagan
Chief Technology Officer
linkedin
Company data provided by crunchbase