Senior Deep Learning Software Engineer

Job Summary

Joining a diverse and ambitious team, the full-time Senior Deep Learning Software Engineer will design and build PyTorch components for efficient operation on supercomputers, collaborating with both internal teams and the PyTorch community in a remote or onsite capacity.

Key responsibilities

• Design and build PyTorch components that run efficiently on supercomputers with thousands of GPUs
• Collaborate with hardware and software teams to enhance GPU performance in PyTorch
• Develop and support production AI solutions for enterprise customers and partners

Required qualifications

• BS in Computer Science or Engineering (or equivalent experience)
• 3+ years of professional experience in deep learning
• Proficiency in C++ programming
• Strong understanding of systems software and interfaces
• Experience with Thread and Distributed Parallel Programming