Senior PyTorch DL Software Engineer - Equity & Benefits

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

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking a Senior Deep Learning Software Engineer focused on PyTorch to design and implement scalable components for AI workloads. You will work across teams to enhance GPU efficiency and contribute to production AI solutions for enterprise customers and partners.

The role requires strong C++ proficiency, deep learning experience, and a track record in large software projects, with collaboration across international PyTorch communities.

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 thousands of GPUs.
• Collaborate with NVIDIA’s hardware and software teams to improve 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

Multithreading

Distributed parallelism

Systems software

Communication

Large projects

Education

BS in Computer Science or Engineering

Tools

CUDA programming

Python

Open source

Job description

NVIDIA is seeking a Senior Deep Learning Software Engineer focused on PyTorch to design and implement scalable components for AI workloads. You will work across teams to enhance GPU efficiency and contribute to production AI solutions for enterprise customers and partners.

The role requires strong C++ proficiency, deep learning experience, and a track record in large software projects, with collaboration across international PyTorch communities.