Deep Learning Frameworks Software Manager



Santa Clara, CA, USA
Posted on Saturday, November 4, 2023

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.

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.

What you’ll be doing:

If you have a good understanding for 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.

What we need to see:

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

Ways to stand out from the crowd:

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

The base salary range is 216,000 USD - 414,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.#deeplearning