Senior PyTorch Deep Learning Engineer - GPU Scale

NVIDIA; California (MO); On-site; USD 184,000 - 357,000; Full time

Benefits offered by this job

Equity

Benefits package

Job summary

NVIDIA is seeking a Senior Deep Learning Software Engineer to contribute to PyTorch and the broader AI platform. You will design PyTorch components for large GPU systems, collaborating with hardware and software teams to boost performance and reliability.

The role requires a solid foundation in C++, deep learning, and distributed systems, plus experience delivering large software projects and strong communication skills. Equity and benefits are included.

Qualifications

• 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

Responsibilities

• 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.

Skills

C++ programming

Distributed parallel programming

Large software projects

Strong communication

Education

BS in Computer Science or Engineering

Tools

CUDA Programming

Python

Job description

NVIDIA is seeking a Senior Deep Learning Software Engineer to contribute to PyTorch and the broader AI platform. You will design PyTorch components for large GPU systems, collaborating with hardware and software teams to boost performance and reliability.

The role requires a solid foundation in C++, deep learning, and distributed systems, plus experience delivering large software projects and strong communication skills. Equity and benefits are included.