Senior PyTorch DL Engineer — High-Performance GPU & Equity

NVIDIA; Washington (District of Columbia); On-site; USD 200,000 - 322,000; Full time

Job summary

NVIDIA is seeking a Senior Deep Learning Software Engineer to contribute to PyTorch and related AI tooling. You will help design and implement components for large-scale GPU systems, collaborating across hardware and software teams worldwide.

The role emphasizes deep learning expertise, systems programming in C++, and close partnership with researchers to deliver production-grade AI solutions for enterprise customers.

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.

Job description

NVIDIA is seeking a Senior Deep Learning Software Engineer to contribute to PyTorch and related AI tooling. You will help design and implement components for large-scale GPU systems, collaborating across hardware and software teams worldwide.

The role emphasizes deep learning expertise, systems programming in C++, and close partnership with researchers to deliver production-grade AI solutions for enterprise customers.