PyTorch Engineer

Graphcore; West of England; Hybrid; GBP 85,000 - 140,000; Full time

Benefits offered by this job

Flexible working

Generous leave

Retirement planning support

Phantom equity

On-site barista and free food

Income protection and life assurance

Private medical insurance and dental

Cycle to work

Give as you earn

Job summary

Graphcore is seeking a Software Engineer to help make Graphcore hardware feel native inside ML frameworks. You will design, implement, and optimize features that connect Graphcore accelerators with PyTorch, Triton and related technologies.

Join a SCRUM team in Frameworks, contribute to upstream PyTorch, and support a hybrid Python/C++ extension. You will solve real engineering problems, ship tooling, and improve performance and developer experience for ML researchers.

Qualifications

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

Responsibilities

• Design, implement, optimise and support PyTorch-related features across PyTorch, Triton and related technologies.
• Contribute to upstream PyTorch and support a hybrid Python and C++ extension.
• Help integrate compiler backend capabilities with torch.compile and improve performance and developer experience.

Skills

Strong core software engineering

Experience developing Python libraries

Experience building performant C++

Understanding of linear algebra / FP /

Git proficiency

Open-source experience

Job description

About The Job

Help make Graphcore hardware feel native inside the ML frameworks engineers use every day.

About The Job

Help make Graphcore hardware feel native inside the ML frameworks engineers use every day.

As a Software Engineer in our PyTorch team, you will help build the software that connects Graphcore accelerators with state-of-the-art machine learning frameworks. You will design, implement, optimise and support features across PyTorch, Triton and related technologies.

Your work will help ML engineers and researchers get more from Graphcore hardware, without changing how they want to build. That means better compatibility, stronger performance and cleaner developer experiences.

You will contribute to upstream PyTorch, support our hybrid Python and C++ extension, and help integrate compiler backend capabilities with torch.compile. You will learn fast, solve real engineering problems and build expertise in a large open-source ecosystem.

The team and culture

You will join the PyTorch team within Frameworks, working in a SCRUM team that ships features, reviews code, writes documentation and supports users. Work happens through clear ownership, practical decisions and close communication across software, AI/ML, research and hardware teams.

The team values maintainable code, thoughtful design and curiosity about unfamiliar systems. Engineers are trusted to investigate, speak up and improve the codebase as they learn.

This is a team for people who like open-source communities, complex technical problems and visible impact. You will move quickly, but with care for users and long-term quality.

What We're Looking For

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

Benefits

• Flexible working: Balance your work and personal life with greater flexibility
• Generous leave: Take time to rest, recharge and enjoy life outside of work
• Retirement planning support: Up to 5% matched pension
• Phantom equity: Share in Graphcore's success
• Workplace experience: Enjoy thoughtfully designed office spaces for collaboration, with free food and an on-site barista to support your day
• Peace of mind protection: Income protection and life assurance to provide financial security for you and your loved ones
• Flexible benefits: Tailor your benefits package with a choice of additional options, including private medical insurance and dental cover
• Optional benefits: Dental cover, health cash plan, private medical insurance, cycle to work scheme, give as you earn

We welcome people from all backgrounds and experiences and are committed to building an inclusive environment where everyone can do their best work.

We're an equal opportunity employer and recognise that everyone brings different strengths and perspectives. If you need any adjustments during the interview process, just let us know - we're happy to support you.

Join the team at Graphcore

Graphcore is one of the world's leading innovators in Artificial Intelligence compute. It is developing hardware, software and systems infrastructure that will unlock the next generation of AI breakthroughs and power the widespread adoption of AI solutions across every industry.

As part of the SoftBank Group, Graphcore is a member of an elite family of companies responsible for some of the world's most transformative technologies. Together, they share a bold vision: to enable Artificial Super Intelligence and ensure its benefits are accessible to everyone.

Graphcore's teams are drawn from diverse backgrounds and bring a broad range of skills and perspectives. A melting pot of AI research specialists, silicon designers, software engineers and systems architects, Graphcore brings together deep expertise to solve complex problems and deliver meaningful progress in AI compute.

If you want to help shape how PyTorch runs on Graphcore hardware, we'd love to hear from you.