PyTorch Engineer

PyTorch Engineer Overview

Graphcore is seeking a talented PyTorch Engineer to join their team in Bristol, UK. The role involves helping to make Graphcore’s hardware feel native within the machine learning frameworks that developers rely on daily. As part of the PyTorch team within the Frameworks department, you will be responsible for designing, implementing, optimizing, and supporting features across PyTorch, Triton, and related open-source technologies. Your work aims to improve compatibility, boost performance, and enhance the developer experience without requiring users to change their existing workflows.

Key responsibilities include contributing to upstream PyTorch development, supporting hybrid Python and C++ extensions, and integrating compiler backend capabilities with torch.compile. You will learn rapidly, solve real-world engineering problems, and build expertise within a large open-source ecosystem. Collaboration is central to this role, with work conducted within a SCRUM team that emphasizes code reviews, documentation, and user support. The team values maintainable code, thoughtful design, curiosity about new systems, and encourages engineers to investigate and improve the codebase actively. This environment is ideal for those who enjoy open-source communities, tackling complex technical challenges, and making a visible impact.

What We’re Looking For

• Strong core software engineering skills and a collaborative attitude.
• Experience developing Python libraries and working professionally with Python.
• Proficiency in building high-performance C++ applications in a commercial setting.
• Understanding of linear algebra, floating-point arithmetic, or numerical analysis relevant to machine learning hardware.
• Confidence in using Git and participating in code reviews within a team environment.
• Willingness to explore and understand large, complex open-source codebases independently and collaboratively.

Benefits

• Flexible working arrangements to support work-life balance.
• Generous leave policies for rest and personal time.
• Retirement support with up to 5% pension matching.
• Opportunity to share in company success through phantom equity schemes.
• Modern office spaces designed for collaboration, with free food and an on-site barista.
• Financial security benefits including income protection and life insurance.
• Option to tailor benefits with private medical insurance, dental cover, and other perks.
• Additional optional benefits such as health cash plans, cycle-to-work schemes, and charitable giving options.

Work Pattern

Not explicitly specified in the job posting.

Sponsorship Information

The company states that visa sponsorship is available for this role.

Degree Requirement: Not Specified

Visa Sponsorship May be