Senior Deep Learning Software Engineer



Software Engineering
Multiple locations
Posted on Thursday, December 28, 2023

We are now looking for a Senior Deep Learning Software Engineer!

At NVIDIA, we are pushing the boundaries of what is possible in AI. To advance the future of NVIDIA's hardware and software, we are seeking a talented Software Engineer who will deliver innovative quantization and sparsity algorithms into our flagship products. In this role, you will rigorously explore, implement, and evaluate new concepts, bridging the realms of hardware and software while setting new benchmarks in AI acceleration. Your work will play a crucial role in influencing our flagship GPU and software roadmaps, enabling major customer wins, and advancing our mission to fulfill the promise and future of AI.

What You'll Be Doing:

  • Implementing, optimizing, and benchmarking state-of-the-art sparsity and quantization algorithms on real-world AI models running on ground breaking hardware.

  • Collaborate closely with hardware, software, and research teams to assess and adopt deep learning algorithmic advancements in sparsity and quantization

  • Provide engineering support to customers using innovative HW and SW approaches

  • Work closely with production teams to develop and integrate the latest deep learning approaches into state-of-the-art libraries and frameworks

What We Need to See:

  • A minimum of a Master's degree in Computer Science, Artificial Intelligence, Applied Math or a related field, or equivalent experience

  • 5+ years of relevant software development experience

  • A strong foundation in deep learning, specifically in quantization, numerics, and sparsity

  • Strong proficiency with modern frameworks such as PyTorch and TensorFlow

  • Familiarity with computer architecture and how it relates to AI algorithms development

  • Previous experience working directly with AI hardware and software development teams.

Ways to Stand Out From the Crowd:

  • Background with benchmarking accuracy and performance of AI models

  • Experience with CUDA programming and GPU performance optimizations

  • Published research or significant contributions to the field of AI, particularly in algorithm development for hardware-software co-design.

As NVIDIA makes inroads into the Datacenter business, our team plays a central role in getting the most out of our exponentially growing datacenter deployments as well as establishing a data-driven approach to hardware design and system software development. We collaborate with a broad cross section of teams at NVIDIA ranging from DL research teams to CUDA Kernel and DL Framework development teams, to Silicon Architecture Teams. As our team grows, and as we seek to identify and take advantage of long term opportunities, our skill-set needs are expanding as well.

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 176,000 USD - 333,500 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.