Senior Deep Learning Frameworks Sustaining Engineer

Key Skills

Aartificial intelligence, CUDA kernels, Deep Learning, GitLab, Machine learning techniques, OpenCL, PyTorch, TensorFlow

Job Description

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!

The base salary range is 148,000 USD - 276,000 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.