Description

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

This role is hybrid.

Responsibilities

• Collaborate with 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.
• Develop large-scale software projects using C++ and distributed parallel programming.
• Contribute to the open source community and optimize deep learning compilers.

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.
• Familiarity with PyTorch.
• 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