Make PyTorch work brilliantly on hardware built for the next era of machine intelligence.
As a Senior Software Engineer in our PyTorch team, you will help Graphcore hardware feel natural to ML engineers and researchers. You will design, optimise and support the software that connects our accelerators with widely used frameworks.
Your work will shape upstream PyTorch contributions, custom kernels and compiler backend integration with torch.compile. The goal is simple: better compatibility, stronger performance and an easier developer experience.
You will work across Python, C++ and complex open source codebases. You will also help coordinate technical tasks, review code and improve how the team builds.
This is a role for engineers who want deep ownership and real technical stretch. You will grow expertise in PyTorch while influencing systems used by ambitious ML teams.
The team and cultureYou will join the PyTorch team within Frameworks, working in a SCRUM team that ships features, fixes issues and improves performance. Work happens through open discussion, code review and close contact with software, AI/ML, research and hardware teams.
Engineers are trusted to take responsibility for complex, open-ended tasks. Decisions are shaped by technical evidence, user impact and healthy challenge.
The team values clarity, learning and practical progress. You will mentor others, speak up when quality matters and help improve how the team works.
What we’re looking for· Strong software engineering skills across complex systems.
· Experience building in Python and performant C++ in a commercial setting.
· Confidence with linear algebra and floating-point arithmetic.
· Strong git skills, including code review in a commercial engineering team.
· Ability to investigate unfamiliar, large or open source codebases and write clear technical documentation.
· Experience in at least one of ML kernels, compiler development, ML frameworks or computationally intensive engineering.
· Flexible working: Balance your work and personal life with greater flexibility
· Comprehensive healthcare and dental cover: Premium medical and dental insurance, with the option to extend cover to your family
· Phantom equity: Share in Graphcore’s success
· Fitness and wellbeing support: A fully funded Medicover gym card, giving you access to a wide range of gyms, fitness classes and leisure centres
· Financial protection: Death and disability insurance, providing financial security for you and your loved ones
· Retirement savings: Build your financial future, with a 4% employer contribution from Graphcore, alongside your own contributions
· Employee Assistance Programme (EAP): Confidential advice and support for you and your family across a range of personal, financial and wellbeing matters
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.
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 shape how PyTorch runs on Graphcore technology, we’d love to hear from you. Apply now to help build the software foundations for the next generation of AI.
Skills Required
- Demonstrable strong software engineering skills
- Experience of Python development
- Experience of developing performant C++ applications in a commercial setting
- At least one of: experience of writing ML kernels; compiler development; experience of using and/or development of ML frameworks; experience in a computationally intensive engineering field
- Demonstrable knowledge of AI/ML
- Understanding of computing architectures
- Experience of maintaining and supporting a complex code base
- Experience in profiling / optimising high performance code
What the Team is Saying
What We Do
At Graphcore, we’re building the future of AI compute. We’re a team of semiconductor, software and AI experts, with deep experience in creating the complete AI compute stack - from silicon and software to infrastructure at datacenter scale. As part of the SoftBank Group, backed by significant long-term investment, we are delivering key technology into the fast-growing SoftBank AI ecosystem. To meet the vast and exciting AI opportunity, Graphcore is expanding its teams around the world. We are bringing together the brightest minds to solve the toughest problems, in a place where everyone has the opportunity to make an impact on the company, our products and the future of artificial intelligence.
Why Work With Us
Our team is at the forefront of the machine intelligence revolution, enabling innovators from all industries to build AI-native products to expand human potential. What we do at Graphcore really makes a difference.
Gallery
Graphcore Offices
Hybrid Workspace
Employees engage in a combination of remote and on-site work.
At Graphcore, we value wellbeing and flexibility to support a healthy work/life balance. Our hybrid approach encourages office-based colleagues to work onsite three days a week, with trusted flexibility built on trust and transparency for everyone.
