Software Manager, Deep Learning Frameworks

Key Skills

Aartificial intelligence, C Programming, C++, CUDA/GPU programming, Deep Learning, JAX framework, Machine learning techniques, Python Programming, PyTorch, TensorFlow

Job Description

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.

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.