Senior PyTorch Deep Learning Engineer — Scalable AI

NVIDIA; Washington; On-site; USD 184,000 - 357,000; Full time

Benefits offered by this job

Equity

Comprehensive benefits

Job summary

NVIDIA is seeking a Senior Deep Learning Software Engineer, PyTorch, to design and implement PyTorch components that run on large-scale GPU systems. You will collaborate with cross-functional teams to optimize GPU performance and deliver production-ready AI solutions for enterprise customers.

The role requires a strong C++ background, deep learning experience, and proven ability to work on large software projects. Equity and comprehensive benefits are available.

Qualifications

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

Skills

C++ programming

Thread Programming

Distributed Parallel Programming

Large software projects

Communication skills

Education

BS in Computer Science or Engineering

Tools

CUDA Programming

Python

Job description

NVIDIA is seeking a Senior Deep Learning Software Engineer, PyTorch, to design and implement PyTorch components that run on large-scale GPU systems. You will collaborate with cross-functional teams to optimize GPU performance and deliver production-ready AI solutions for enterprise customers.

The role requires a strong C++ background, deep learning experience, and proven ability to work on large software projects. Equity and comprehensive benefits are available.