Make PyTorch feel native, powerful and production ready on Graphcore AI compute.
You will build the software layer that connects Graphcore accelerators with leading machine learning frameworks. Your work will shape how engineers run real workloads at scale.You will improve performance, design new features and support users pushing hardware to its limits. You will work deeply in PyTorch, with scope across other frameworks as the ecosystem evolves.
You will maintain core code, review changes and contribute to documentation and upstream communities. You will also help manage complexity, quality and delivery across a fast moving codebase. This role offers deep technical ownership and influence over direction. You will help unlock the next generation of AI breakthroughs through better software.
The team and cultureThe Frameworks team connects Graphcore hardware with tools engineers already use, including PyTorch, Triton, JAX and TensorFlow. Work is grounded in real use cases, performance goals and developer experience.
Teams operate through Scrum, with clear ownership from design to delivery, testing and support. Engineers are expected to speak up, challenge ideas and move work forward.
Decisions are based on evidence, code quality and user impact. You will collaborate across teams and with open source communities to deliver reliable, high performance software.
What we’re looking for· Strong software engineering skills, with the judgement to manage quality, complexity and delivery.
· Experience developing in Python.
· Experience building performant C++ applications in a commercial setting.
· Experience in at least one of ML kernels, compiler development, ML frameworks or computationally intensive engineering.
· Ability to coordinate complex technical tasks with cross-team dependencies.
· Clear communication, ownership and curiosity about high performance software.
While we have outlined a set of requirements, we value transferable skills and diverse experiences. If you meet most of our essential criteria, we encourage you to submit an application and showcase how your background makes you a strong candidate.
Benefits· 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 developers use PyTorch on advanced AI hardware, we’d love to hear from you. Apply now to be part of the journey.
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
Graphcore Compensation & Benefits Highlights
-
Healthcare Strength — Private medical insurance, comprehensive dental/vision, a health cash plan, and mental wellbeing support are listed as standard, with day-one coverage and Spring Health/EAP support noted for the U.S. Life assurance and income protection add further protection for serious illness or injury.
-
Leave & Time Off Breadth — Flexible working hours and “unlimited annual leave” are explicitly stated. U.S. materials also note flexible PTO and 11 paid holidays.
-
Retirement Support — Retirement programs include a U.K. pension match up to 5% and a U.S. 401(k) with a 100% company match up to 6% (with a year-end true-up). These provisions signal a meaningful employer contribution toward long-term savings.
Graphcore Insights
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.
