Software Manager, Deep Learning Frameworks
Job Description
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 220,000 USD - 419,750 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.