Software Manager, Deep Learning Frameworks

World leader in accelerated computing, pioneering AI and digital twins technology transforming major industries.
$224,000 - $425,500
Machine Learning
Staff Software Engineer
In-Person
5+ years of experience
AI

Description For Software Manager, Deep Learning Frameworks

NVIDIA is seeking a Software Manager for their GPU-accelerated Optimized Deep Learning Frameworks team. This role involves leading a team of software engineers developing core deep learning system software and algorithms. The position focuses on implementing new algorithms, developing compiler features, integrating cuDNN and CUDA features, and supporting NVIDIA GPUs. The ideal candidate will have strong leadership experience and deep understanding of machine learning frameworks.

The role offers an opportunity to work with cutting-edge technology in deep learning and AI, collaborating with academic and commercial groups worldwide. You'll be at the forefront of GPU-powered innovations, from image classification to large language models. The position requires expertise in C/C++ programming, parallel computing, and machine learning frameworks, with additional value placed on GPU architecture knowledge and CUDA programming experience.

As a leader at NVIDIA, you'll be responsible for defining project goals, managing team development efforts, and maintaining direct contact with the open source community. The company offers competitive compensation, including a substantial base salary range and equity benefits. NVIDIA is known for its forward-thinking approach and commitment to innovation, making it an ideal environment for those passionate about advancing deep learning technology.

Working at NVIDIA means joining one of technology's most desirable employers, where creativity and autonomy are highly valued. The role combines technical expertise with leadership responsibilities, offering a unique opportunity to influence the future of deep learning software that will be used globally.

Last updated a month ago

Responsibilities For Software Manager, Deep Learning Frameworks

  • Lead and mentor forward-thinking engineers
  • Own related activities and interactions with teams across NVIDIA
  • Partner with key internal partners on priority alignment
  • Define project goals and scope
  • Direct contact with the open source community
  • Manage team's development effort
  • Develop core deep learning system software and algorithms
  • Performance tuning and analysis

Requirements For Software Manager, Deep Learning Frameworks

Python
Linux
  • 5+ years of overall relevant work experience leading a software product
  • 2+ years of managing a team
  • Background in Applied Math or Computer Science
  • Excellent C/C++ programming and software design skills
  • Solid understanding of parallel execution environment
  • Experience with machine learning frameworks (TensorFlow, PyTorch, JAX, or MXNet)
  • Experience in open source software development
  • Masters or PhD in Deep Learning, Artificial Intelligence or related field or equivalent experience

Benefits For Software Manager, Deep Learning Frameworks

Equity
  • Equity

Interested in this job?

Jobs Related To NVIDIA Software Manager, Deep Learning Frameworks

Machine Learning Software Platform Architect

Senior ML Platform Architect role at NVIDIA focusing on LLM infrastructure for chip design

Gen AI Product Evangelist Engineer, Retail

Senior technical role combining AI expertise with developer advocacy to create educational content and demos for NVIDIA's retail-focused Gen AI applications.

Manager, Neural Reconstruction Software

Lead NVIDIA's Neural Reconstruction Software team, developing cutting-edge AI technologies for computer vision and robotics applications.

Senior Manager, Visual AI Microservices

Lead the development of Visual AI Microservices at NVIDIA, managing teams building cutting-edge computer vision and AI services for developer enablement.

Manager, Artificial Intelligence Algorithms

Lead AI algorithms team at NVIDIA, managing deep learning software development and optimization for GPU-accelerated platforms.