Senior PyTorch & Deep Learning Software Engineer

NVIDIA; New York (NY); 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 to contribute to PyTorch and build scalable tooling for AI across thousands of GPUs. You will collaborate with multiple teams worldwide and engage with the PyTorch community to advance the AI platform.

The role requires a BS in CS/Engineering, 3+ years in deep learning, and strong C++ and systems software skills. Expect opportunities to work on large-scale projects, with equity and benefits in the compensation package.

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 supercomputers with thousands to tens of thousands of GPUs.
• Collaborate with NVIDIA’s 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

Deep learning

C++ programming

Distributed parallel programming

Systems software

Communication skills

Large software projects

Education

BS in Computer Science or Engineering

Tools

CUDA Programming

Python

Job description

NVIDIA is seeking a Senior Deep Learning Software Engineer to contribute to PyTorch and build scalable tooling for AI across thousands of GPUs. You will collaborate with multiple teams worldwide and engage with the PyTorch community to advance the AI platform.

The role requires a BS in CS/Engineering, 3+ years in deep learning, and strong C++ and systems software skills. Expect opportunities to work on large-scale projects, with equity and benefits in the compensation package.