Senior Software Engineer, PyTorch - Deep Learning

Nvidia

Confirmed live today High trust

Hybrid

Quick summary

Work type: Hybrid
Location: Santa Clara, CADCNYWACAMA
Salary: $184,000–$287,500 / yr
Employment: Full-time
Posted: 3 days ago
Freshness: Confirmed live today
Nearby: 99+ roles within 25 mi

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $217k

This role $236k

$162k most similar roles pay here $301k

This role pays more than 68% of similar roles. Most pay $188,675–$246,037 — the shaded band above. At the midpoint, this role pays about $236k versus about $217k for comparable roles.

Based on 240 similar postings.

Employer

About Nvidia

Nvidia is a leading designer of graphics processing units (GPUs) and system-on-chip units, powering gaming, professional visualization, data centers, and artificial intelligence workloads. Industry: Semiconductors & AI Computing

Nvidia currently has 892 open roles on FindRole.

Listed pay typically runs $184,000–$287,500 across 870 roles with salary data.

Most-posted roles

View all roles at Nvidia

At a glance

TL;DR · Senior Software Engineer, PyTorch - Deep Learning

Role Posting Log in to save

As a Senior Software Engineer, PyTorch - Deep Learning, you will join an ambitious team focused on the development of the PyTorch deep learning framework and the broader AI platform. You will design and build PyTorch components optimized for supercomputers with thousands of GPUs while collaborating with hardware and software teams to improve overall GPU performance. Your daily work involves building production AI solutions for enterprise customers and improving tools for internal applied researchers. The role requires proficiency in C++ programming, a strong understanding of systems software and interfaces, and experience in thread and distributed parallel programming. Preferred skills include CUDA programming, Python, and familiarity with deep learning compilers and modeling trends. You will solve complex problems related to scaling AI applications across massive GPU clusters within the core deep learning infrastructure space.

What does a Software Engineer earn in California?

Median $214000 from 777 postings across 68 companies.

See salary data

Skills

PyTorch C++ Python CUDA Deep Learning Distributed Parallel Programming Systems Software Deep Learning Compilers Artificial Intelligence

What you'll do

Design and build PyTorch components that run efficiently on supercomputers with thousands of GPUs.
Collaborate with hardware and software teams to improve overall GPU performance in PyTorch.
Develop and support production AI solutions for enterprise customers and partners.
Improve internal tools used by applied researchers.
Contribute to the development of the global PyTorch deep learning framework.
Implement distributed parallel programming and systems software interfaces.

What we're looking for

BS in Computer Science or Engineering (or equivalent experience).
3+ years of professional experience in deep learning.
Proficiency in C++ programming.
Strong understanding of systems software and interfaces.
Demonstrated experience with Thread and Distributed Parallel Programming.
Demonstrated background developing large software projects.
Strong verbal and written communication skills.
Experience with open source, deep learning compilers, modeling trends, CUDA, or Python (preferred).