Senior PyTorch Deep Learning Engineer | Scalable AI Platforms

NVIDIA; Santa Clara (CA); On-site; USD 200,000 - 322,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, build and optimize components that run on thousands of GPUs. You will collaborate across NVIDIA's hardware and software teams to enhance PyTorch performance and deliver production AI solutions for enterprise customers.

The role requires a BS in CS/Engineering, 12+ years in deep learning, and strong C++ and systems programming skills. CUDA and Python experience are a plus, with equity and benefits included.

Qualifications

• BS in Computer Science or Engineering (or equivalent).
• 12+ 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.

Responsibilities

• Design and build PyTorch components for GPUs at scale.
• Collaborate with hardware and software teams to improve GPU performance in PyTorch.
• Design, build and support production AI solutions for enterprise customers and partners.
• Work with internal applied researchers to improve AI tools.

Skills

Deep learning

C++ programming

Distributed parallel programming

Large software projects

Communication skills

Education

BS in CS or Engineering

Tools

CUDA programming

Python

PyTorch

Job description

NVIDIA is seeking a Senior Deep Learning Software Engineer focused on PyTorch to design, build and optimize components that run on thousands of GPUs. You will collaborate across NVIDIA's hardware and software teams to enhance PyTorch performance and deliver production AI solutions for enterprise customers.

The role requires a BS in CS/Engineering, 12+ years in deep learning, and strong C++ and systems programming skills. CUDA and Python experience are a plus, with equity and benefits included.