Description

You will design and build PyTorch components for efficient execution on supercomputers equipped with thousands of GPUs.

Responsibilities

• Collaborate with NVIDIA’s hardware and software teams to enhance overall GPU performance within PyTorch.
• Design, build, and support production AI solutions for enterprise customers and partners.
• Work with internal applied researchers to improve existing AI tools.

Required Skills

• 3+ years of professional experience in deep learning.
• Proficiency in C++ programming.
• Strong understanding of systems software and interfaces.
• Demonstrated experience with Thread and Distributed Parallel Programming.
• Experience developing large-scale software projects.
• Familiarity with PyTorch.
• Experience with C++.
• Demonstrated background working with multi-disciplinary teams.

Preferred Skills

• Contributions and participation in the open source community.
• Familiarity with deep learning compilers.
• Background with CUDA Programming and Python.

Key Skills

Education

Any Graduate