Software Manager, Deep Learning Frameworks

Details

NVIDIA is hiring software leaders for its GPU-accelerated Optimized Deep Learning Frameworks team. We work with academic and commercial groups around the world who are using GPUs to power a revolution in deep learning, enabling breakthroughs in problems from image classification to recommender systems to large language models. Join the NVIDIA team which is building deep learning software that will be used by the entire world.

Responsibilities

In this role, you will be leading a team of software engineers that are developing core deep learning system software and algorithms. The scope of these efforts ranges from implementation of new algorithms like mixed precision, development and integration of deep learning compiler features, integration of the latest cuDNN and CUDA features, performance tuning and analysis, and support for the latest and greatest NVIDIA GPUs.

Requirements

If you have a good understanding of deep learning and a strong algorithmic background with exposure to computer architecture and performance and leadership experience, then this role may be a great one for you! You'll have the opportunity to lead and mentor forward-thinking engineers and be responsible for ownership of related activities and interactions with teams across NVIDIA while partnering with key internal partners on priority alignment across relevant teams for roadmap development of highly optimized novel and powerful numerical, analytics, and deep learning algorithms, and associated R&D duties.

Required Qualifications

• 5+ years of overall relevant work experience leading a software product.
• 2+ years of managing a team.
• Ability to define project goals and scope, direct contact with the open source community, and manage the team’s development effort.
• Background in Applied Math or Computer Science
• Excellent C/C++ programming and software design skills, including debugging, performance analysis, and test design. Python experience also highly desired.
• Solid understanding of a parallel execution environment.
• Prior experience with machine learning algorithms and frameworks (TensorFlow, PyTorch, JAX, or MXNet).
• Experience in open source software development efforts.
• Masters or PhD in Deep Learning, Artificial Intelligence or related field or equivalent experience.

Preferred Qualifications

• Knowledge of GPU architectures.
• CUDA programming experience.

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!