Senior Deep Learning Software Engineer, cuDNN

at Nvidia
📍 Santa Clara, United States
USD 184,000-356,500 per year
SENIOR
✅ On-site

SCRAPED

Used Tools & Technologies

Not specified

Required Skills & Competences ?

Python @ 3 Algorithms @ 4 Data Structures @ 4 Machine Learning @ 3 Data Science @ 7 API @ 4 CUDA @ 3 GPU @ 4

Details

We are looking for a Senior Deep Learning Software Engineer to join the cuDNN team to design, build, and ship GPU-accelerated primitives for deep neural networks. The role focuses on developing production-quality software as part of NVIDIA's AI software stack, including support for cutting-edge large language models, and working closely with GPU architects and cross-collaborative teams.

Responsibilities

  • Develop production-quality software that ships as part of NVIDIA's AI software stack, including support for large language models.
  • Analyze performance of important workloads, tune existing software, and propose improvements for future software.
  • Collaborate with deep learning software engineers and GPU architects across applications such as large language models, autonomous driving, computer vision, and recommender systems.
  • Contribute across the codebase: API design, software architecture, testing, and GPU kernel development.
  • Mentor and guide junior engineers on the team.

Requirements

  • M.S. Degree in Computer Science, Electrical Engineering (or similar), or equivalent experience.
  • 4+ years of relevant work or research experience.
  • Strong programming skills in C/C++ development.
  • Work experience with CUDA development and familiarity with Python.
  • Good understanding of linear algebra.
  • Familiarity with the latest trends in machine learning, especially large language models.
  • Experience designing high-level software architecture.
  • Excellent problem solving skills, including applications of algorithms and data structures.
  • Experience with performance analysis, profiling, and code optimization.
  • Ability to work independently, define project goals and scope, and lead development efforts.

Ways to stand out

  • GPU programming and optimization expertise (e.g., CUDA or OpenCL).
  • Practical experience with machine learning, especially deep learning.
  • Strong experience with data science, statistical analysis, and visualization.
  • Previous work on large complex codebases with many developers (libraries, compilers, or system software).
  • Track record of identifying and incorporating new technologies into development flows.

Compensation & Benefits

  • Base salary ranges (location, experience, and peer pay dependent):
    • Level 4: 184,000 USD - 287,500 USD
    • Level 5: 224,000 USD - 356,500 USD
  • Eligible for equity and benefits.

Additional Information

  • Applications for this job will be accepted at least until August 5, 2025.
  • NVIDIA is an equal opportunity employer and is committed to fostering a diverse work environment.