Job Description

NVIDIA is seeking a Senior Deep Learning Software Engineer for their cuDNN team to develop high-performance software for deep neural networks. The role involves collaborating with cross-functional teams, optimizing software performance, and mentoring junior engineers in a rapidly evolving AI landscape.

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 in 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 your own development effort.

Responsibilities

Develop production-quality software that ships as part of NVIDIA's AI software stack, including cutting edge large language model support.
Analyze the performance of important workloads, tuning our current software, and proposing improvements for future software.
Work with cross-collaborative teams of deep learning software engineers and GPU architects to innovate across applications like large language models, autonomous driving, computer vision, and recommender systems.
Adapt to the constantly evolving AI industry by being agile and excited to contribute across the codebase, including API design, software architecture, testing, and GPU kernel development.
Mentoring junior engineers on the team.

Benefits

Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.