Senior Software Engineer, PyTorch - Deep Learning

Position Overview

NVIDIA is seeking a Senior Software Engineer, PyTorch - Deep Learning to design and build advanced tools and components for the PyTorch deep learning framework. Working with diverse, international teams and the open source community, you will improve GPU performance across large-scale supercomputers and create production AI solutions.

Locations: US, CA, Santa Clara; US, NC, Remote; US, NY, Remote; US, CA, Remote. Time Type: Full time.

Key Responsibilities

• Design and build PyTorch components that run efficiently on supercomputers with 1,000s–100,000s of GPUs.
• Collaborate with NVIDIA’s hardware and software teams to enhance 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 AI tools.

Required Qualifications

• BS in Computer Science or Engineering (or equivalent experience).
• 3+ years of professional experience in deep learning.
• Proficient with C++ programming.
• Strong understanding of systems software and interfaces.
• Experience with thread and distributed parallel programming.
• Proven track record in developing large software projects.
• Excellent verbal and written communication skills.

Preferred Qualifications

• Contributions and active participation in the open source community.
• Familiarity with deep learning compilers and modeling trends.
• Background with CUDA programming as well as Python.
• Experience working with multi-disciplinary teams.

Benefits & Perks

• Highly competitive salaries with a comprehensive benefits package.
• Base Salary Range:Level 3: $148,000 – $235,750; Level 4: $184,000 – $287,500 (salary determined based on location, experience, and comparable employee pay).
• Eligibility for equity participation.
• Flexible work options including hybrid and remote opportunities.
• Application deadline: Accepted at least until November 25, 2025.

NVIDIA is committed to fostering a diverse work environment and is proud to be an equal opportunity employer.