Senior Deep Learning Frameworks Sustaining Engineer

NVIDIA

NVIDIA

Santa Clara, CA, USA
Posted on Friday, December 15, 2023

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 work 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.

  • Solid understanding of the tools used in building software (Make, Docker, Bazel), packaging systems (Debian, pip, npm, etc.)

  • 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.

What we would like to see:

  • 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!

The base salary range is 144,000 USD - 270,250 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.