Senior Software Engineer, PyTorch - Deep Learning

Nvidia

Confirmed live yesterday High trust

Remote

Quick summary

Work type: Remote
Location: Santa Clara, CADCNYWACAMA
Employment: Full-time
Posted: 15 days ago
Freshness: Confirmed live yesterday
Nearby: 99+ roles within 25 mi

Market check

Salary context

How this pay compares to similar roles

Similar $216k

$168k $273k

below market most similar roles pay here above market

This listing doesn't post a salary. Most similar roles pay $186,300–$246,150.

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 1626 open roles on FindRole.

Listed pay typically runs $184,000–$287,500 across 1115 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 to design and build tools used by AI engineers to develop and deploy applications scalable across thousands of GPUs. You will design and build PyTorch components for supercomputers, collaborate with hardware and software teams to improve GPU performance, and support production AI solutions for enterprise customers. Your daily work involves collaborating with internal applied researchers and the international PyTorch community. The role requires proficiency in C++ programming, systems software, and interfaces, alongside experience in Thread and Distributed Parallel Programming. You will utilize knowledge of deep learning compilers, CUDA programming, and Python to solve complex problems within the deep learning framework and the broader AI platform.

What does a Software Engineer earn in California?

Median $214500 from 1223 postings across 76 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 for supercomputers with thousands of GPUs.
Improve overall GPU performance in PyTorch by collaborating with hardware and software teams.
Design, build, and support production AI solutions for enterprise customers and partners.
Improve AI tools for internal applied researchers.
Develop large-scale software projects using C++ and distributed parallel programming.
Contribute to the PyTorch open-source community and global AI platform development.

What we're looking for

BS in Computer Science or Engineering or equivalent experience.
12+ 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.
Background with CUDA Programming and Python (preferred).