NVIDIA is seeking a Senior Deep Learning Software Engineer to help build tools for designing, developing, and deploying AI applications across large-scale GPU systems. The role focuses on developing PyTorch components, improving GPU performance, supporting production AI solutions, and collaborating with hardware, software, research, and open-source teams.

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

Senior Software Engineer, PyTorch - Deep Learning

Key skills

About this role

Responsibilities: