Senior PyTorch Deep Learning Engineer (GPU AI)

NVIDIA Corporation; Santa Clara (CA); On-site; USD 184,000 - 357,000; Full time

Job summary

NVIDIA Corporation is seeking a Senior Deep Learning Software Engineer for PyTorch to design and build scalable PyTorch components. The role involves collaboration with NVIDIA’s hardware and software teams to boost GPU performance and to deliver production AI solutions for enterprise customers and partners.

You will work with internal researchers to advance AI tools, contribute to PyTorch, and help shape the AI platform across NVIDIA’s global teams.

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 large-scale GPUs.
• Collaborate with 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

Deep learning

Distributed systems

Large software projects

Communication skills

Education

BS in Computer Science or Engineering

Tools

CUDA Programming

Python

Job description

NVIDIA Corporation is seeking a Senior Deep Learning Software Engineer for PyTorch to design and build scalable PyTorch components. The role involves collaboration with NVIDIA’s hardware and software teams to boost GPU performance and to deliver production AI solutions for enterprise customers and partners.

You will work with internal researchers to advance AI tools, contribute to PyTorch, and help shape the AI platform across NVIDIA’s global teams.