Senior Software Engineer, PyTorch – Deep Learning

NVIDIA

Senior Software Engineer working on AI tools for NVIDIA's deep learning platform. Collaborates on GPU performance and supports enterprise AI solutions.

Posted 7/3/2026full-timeSanta Clara • California, New York, North Carolina • 🇺🇸 United StatesSenior💰 $184,000 - $356,500 per yearWebsite

Tech Stack

Tools & technologies

PyTorch

About the role

Key responsibilities & impact

Design and build PyTorch components that run efficiently on supercomputers with 1000s-100ks of GPUs
Collaborate with NVIDIA’s hardware and software teams to improve the overall GPU performance in PyTorch
Design, build and support production AI solutions used by enterprise customers and partners
Work with internal applied researchers to improve their AI tools

Requirements

What you’ll need

BS in Computer Science or Engineering (or equivalent experience)
3+ years professional experience in deep learning
Proficient with 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

Benefits

Comp & perks

Comprehensive benefits package
Equity

ATS Keywords

✓ Tailor your resume

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

Deep LearningC++ ProgrammingThread ProgrammingDistributed Parallel ProgrammingSystems SoftwareLarge Software ProjectsAI Solutions Development

Soft Skills

Verbal CommunicationWritten Communication