PyTorch Engineer

📋 Description

• Help make Graphcore hardware feel native inside ML frameworks
• Build software connecting Graphcore accelerators with ML frameworks (PyTorch, Triton)
• Design, implement, optimise and support PyTorch features
• Contribute to upstream PyTorch and integrate torch.compile
• Work in SCRUM, cross-team collaboration

🎯 Requirements

• Strong core software engineering skills
• Experience developing Python libraries
• Experience building performant C++ applications in a commercial setting
• Understanding of linear algebra, floating-point arithmetic or numerical analysis
• Proficient with Git and code reviews
• Willingness to explore large open-source codebases independently

🎁 Benefits

• Flexible working
• Generous leave
• Retirement/pension up to 5% matched
• Phantom equity
• Office spaces with free food and on-site barista
• Income protection and life insurance