Senior Deep Learning Frameworks Sustaining Engineer

NVIDIA Corporation; Santa Clara (CA); On-site; USD 152,000 - 288,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA Corporation is seeking passionate software engineers for the Deep Learning Frameworks Sustaining Engineering team in Santa Clara, CA. You will back-port changes, track open-source dependencies, and help ensure stable, secure enterprise releases for TensorFlow, PyTorch, and TensorRT.

Join a world-class team delivering long-term support for NVIDIA AI Enterprise customers, contributing to open source codebases, and collaborating across teams to patch vulnerabilities and keep dependencies

Qualifications

• Bachelor of Science in Computer Science or related field or equivalent experience.
• 5+ years of relevant software development experience.
• Excellent C/C++ programming and software design, debugging and open source integration; Python experience helpful.
• Experience with building tools (Make, Docker, Bazel); packaging systems (Debian, pip, npm); CI/CD (GitLab CI/Jenkins).
• Prior experience with ML algorithms and frameworks (TensorFlow, PyTorch, MXNet).
• Ability to work independently, contribute to release stability, and communicate status clearly.

Responsibilities

• Back-port changes from the mainline; keep track of open source dependency changes; ensure latest stable dependencies for enterprise products.
• Contribute changes to support timely Long Term Support releases for TensorFlow, PyTorch, and TensorRT; fix customer-reported bugs.
• Coordinate with teams to patch OSS dependencies for security vulnerabilities for NVIDIA AI Enterprise subscribers.

Skills

C/C++

Python

Open Source

Docker

Bazel

GitLab CI

Make

TensorFlow

PyTorch

MXNet

Education

B.S. in Computer Science or related field

Tools

Make

Docker

Bazel

GitLab CI

Debian

pip

npm

Job description

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.

You will also be eligible for equity and benefits.

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.