Senior PyTorch Deep Learning Engineer - GPU-Scale Impact

NVIDIA; Massachusetts; 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 build components that run on thousands of GPUs and contribute directly to PyTorch, the leading deep learning framework.

You will collaborate with hardware and software teams to improve GPU performance in PyTorch, design and support production AI solutions for enterprise customers, and work with researchers to advance AI tools across NVIDIA's platform.

Qualifications

• BS in Computer Science or Engineering or equivalent experience.
• 3+ years professional deep learning experience.
• Proficient with C++ programming.
• Strong understanding of systems software and interfaces.
• Experience with Thread and Distributed Parallel Programming.
• Experience building large software projects.
• Excellent 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

Distributed parallel programming

Large software projects

Strong communication

Deep learning experience

Python experience

Education

BS in Computer Science or Engineering

Tools

CUDA Programming

PyTorch

Python

Job description

NVIDIA is seeking a Senior Deep Learning Software Engineer, PyTorch, to design and build components that run on thousands of GPUs and contribute directly to PyTorch, the leading deep learning framework.

You will collaborate with hardware and software teams to improve GPU performance in PyTorch, design and support production AI solutions for enterprise customers, and work with researchers to advance AI tools across NVIDIA's platform.