Senior PyTorch DL Engineer — Scale AI on GPUs (Equity)

NVIDIA Corporation; 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 for PyTorch to design and build PyTorch components and collaborate with multiple teams to accelerate GPU performance across NVIDIA’s AI platform.

You will work with enterprise customers and researchers to deploy production AI solutions, contribute to PyTorch, and engage with the international PyTorch community to advance the field.

Qualifications

• BS in Computer Science or Engineering or equivalent experience.
• 12+ 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 GPUs.
• Collaborate with NVIDIA’s hardware and software teams to improve PyTorch GPU performance.
• 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

Systems software

Threading/Distributed programming

Large software projects

Communication

Education

BS in Computer Science or Engineering

Tools

CUDA Programming

Python

Job description

NVIDIA is seeking a Senior Deep Learning Software Engineer for PyTorch to design and build PyTorch components and collaborate with multiple teams to accelerate GPU performance across NVIDIA’s AI platform.

You will work with enterprise customers and researchers to deploy production AI solutions, contribute to PyTorch, and engage with the international PyTorch community to advance the field.