Senior PyTorch DL Engineer — Build Tools at Scale (Equity)

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

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA is seeking a Senior Deep Learning Software Engineer, PyTorch to help design and build PyTorch components that run efficiently on thousands of GPUs. You will collaborate across NVIDIA's hardware and software teams and with the PyTorch community to push the AI platform forward.

You will work with cross-disciplinary teams to develop production AI solutions for enterprise customers, and mentor others while maintaining strong C++ and systems software foundations.

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 GPU clusters.
• 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

Deep learning

Team collaboration

Communication skills

Software development

Education

BS in Computer Science

Tools

CUDA Programming

Python

PyTorch

Job description

NVIDIA is seeking a Senior Deep Learning Software Engineer, PyTorch to help design and build PyTorch components that run efficiently on thousands of GPUs. You will collaborate across NVIDIA's hardware and software teams and with the PyTorch community to push the AI platform forward.

You will work with cross-disciplinary teams to develop production AI solutions for enterprise customers, and mentor others while maintaining strong C++ and systems software foundations.