Taro Logo

Senior Deep Learning Frameworks Sustaining Engineer

NVIDIA is the world leader in accelerated computing, pioneering solutions for AI and digital twins.
$148,000 - $287,500
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI

Job Description

NVIDIA, the world leader in accelerated computing, is seeking a Senior Deep Learning Frameworks Sustaining Engineer to join their team. This role is crucial in maintaining and improving NVIDIA's enterprise AI products, specifically working with frameworks like TensorFlow, PyTorch, and TensorRT.

The position offers a competitive salary range of $148,000 - $287,500 USD (depending on level) plus equity and benefits, reflecting NVIDIA's position as one of technology's most desirable employers. Based in Santa Clara, CA, you'll be part of a team that's driving innovation in deep learning and AI.

As a Senior Engineer, you'll be responsible for maintaining the stability and security of NVIDIA's enterprise AI frameworks. Key responsibilities include back-porting changes, managing dependencies, fixing bugs, and ensuring security compliance. This role requires strong expertise in C/C++ and Python, along with deep understanding of machine learning frameworks and build systems.

The ideal candidate brings 5+ years of software development experience and a strong educational background in Computer Science, Deep Learning, or related fields. You should be comfortable working independently while maintaining effective communication with cross-functional teams. Experience with GPU programming (CUDA) and open source project management would be valuable additions to your skill set.

At NVIDIA, you'll be working at the forefront of AI and deep learning technology, contributing to software that enables breakthroughs in image classification, speech recognition, and natural language processing. The company offers a diverse and inclusive work environment, with a strong commitment to fostering innovation and professional growth.

This role presents an excellent opportunity for an experienced engineer to make significant impacts in the field of AI while working with cutting-edge technology at a leading global company. Join NVIDIA to be part of a team that's shaping the future of accelerated computing and artificial intelligence.

Last updated a day ago

Responsibilities For Senior Deep Learning Frameworks Sustaining Engineer

  • Back-porting changes from the mainline branch
  • Keeping track of open source dependency changes
  • Ensuring latest stable dependencies are used for enterprise products
  • Supporting Long Term Support releases for TensorFlow, PyTorch, TensorRT products
  • Fixing customer reported bugs
  • Integrating bug fixes found in mainline
  • Working with teams to ensure open source dependencies are patched for security vulnerabilities

Requirements For Senior Deep Learning Frameworks Sustaining Engineer

Python
  • Bachelors of Science in Computer Science, Deep Learning, AI, Applied Math, or related field
  • 5+ years of relevant software development experience
  • Excellent C/C++ programming and software design skills
  • Python experience
  • Experience with build tools (Make, Docker, Bazel)
  • Experience with packaging systems (Debian, pip, npm)
  • Experience with Build Systems (Gitlab, CI/Jenkins)
  • Prior experience with machine learning algorithms and frameworks
  • Ability to work independently
  • Strong communication skills

Benefits For Senior Deep Learning Frameworks Sustaining Engineer

Equity
  • Equity
  • Comprehensive benefits package

Related Jobs

Senior Architecture Energy Modeling Engineer

Senior Architecture Energy Modeling Engineer role at NVIDIA focusing on ML-based power modeling and energy efficiency optimization for GPUs, offering $168K-$310K base salary plus equity.

Senior DFX Software Engineer - Machine Learning

Senior DFX Software Engineer role at NVIDIA focusing on machine learning applications in silicon testing, offering $136K-$264.5K salary plus benefits.

Senior Software Engineer, Agentic AI

Senior Software Engineer position at NVIDIA focusing on developing the Agent Intelligence (AIQ) toolkit for enterprise AI applications, requiring 5+ years of Python experience and expertise in LLM frameworks.

Senior Computer Vision System Performance Engineer

Senior Computer Vision System Performance Engineer role at NVIDIA focusing on optimizing computer vision applications and developing hardware-accelerated pipelines.

Senior Deep Learning Systems Engineer, Datacenters

Senior Deep Learning Systems Engineer position at NVIDIA focusing on optimizing datacenter performance for AI applications, offering competitive compensation and opportunity to work with cutting-edge technology.