Software Manager, Deep Learning Frameworks

at Nvidia

📍 Santa Clara, United States

$220,000-419,800 per year

MIDDLE SENIOR
✅ On-site

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Software Development @ 3 Python @ 3 Algorithms @ 3 Machine Learning @ 3 TensorFlow @ 3 Hiring @ 3 Debugging @ 3 PyTorch @ 3 CUDA @ 3

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

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

Benefits

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.