Senior Software Engineer, PyTorch - Deep Learning

Responsibilities

• Design and build PyTorch components for supercomputers with thousands to hundreds of thousands of GPUs
• Collaborate with the company’s hardware and software teams for enhanced GPU performance in PyTorch
• Build and support production AI solutions for enterprise customers
• Work with internal researchers to enhance AI tools

Qualifications

• BS in Computer Science or Engineering (or equivalent experience)
• 3+ years professional experience in deep learning
• Proficient in C++ programming
• Strong understanding of systems software and interfaces
• Experience with Thread and Distributed Parallel Programming
• Background in developing large software projects
• Strong verbal and written communication skills

Benefits

• Eligible for equity
• Hybrid work opportunity
• Access to advanced technology and resources at the company
• Opportunities for professional growth and development
• Involvement in cutting-edge AI projects and contributions to the open-source community

Pay

$184,000 – $356,500 annually

Schedule

Remote - US based candidates only, no visa sponsorship available

Company Information

Our client is an equal opportunity employer. We encourage you to apply even if you don’t meet every qualification—your background could be exactly what this team needs.