Description
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. Apply now to be part of the journey.
Similar jobs
Est. 306,300 PLN
Salary Range: PLN Range: 260,400 - 352,200 Subject to alignment to the responsibilities and duties of the role About the job Make PyTorch work brilliantly on hardware built for the next era of machine intelligence. As a…
Est. 80,000 GBP
About the job Build the framework support that helps AI developers unlock Graphcore hardware. You will help Graphcore accelerators work seamlessly with state-of-the-art ML frameworks, including Triton and PyTorch. Report…
Est. 120,000 PLN
About the job Build the framework support that helps AI developers unlock Graphcore hardware. You will help Graphcore accelerators work seamlessly with state-of-the-art ML frameworks, including Triton and PyTorch. Report…
Est. 95,000 GBP
About the job Build the compute kernels that make next-generation AI hardware perform at its limit. As a Senior Software Engineer, you will create high-performance AI compute libraries for Graphcore’s next-generation har…
Est. 80,000 GBP
About the job Validate the ML stack that turns accelerator hardware into trusted AI performance. This role sits where modern ML models meet Graphcore’s software and hardware stack. You will test, benchmark and validate c…
Est. 65,000 GBP
About the job Build the hardware that turns new AI processors into working systems. As a Hardware Development Engineer, you will design circuit solutions for Graphcore processor systems and validation platforms. You will…
Est. 80,000 GBP
About the job Validate the ML stack that turns accelerator hardware into trusted AI performance. This role sits where modern ML models meet Graphcore’s software and hardware stack. You will test, benchmark and validate c…
Est. 65,000 GBP
About the job Lead the projects that turn advanced hardware designs into products ready for manufacturing. As a Project Manager, you will drive hardware manufacturing projects from planning through production release. Yo…
Est. 80,000 GBP
About the job Validate the ML stack that turns accelerator hardware into trusted AI performance. This role sits where modern ML models meet Graphcore’s software and hardware stack. You will test, benchmark and validate c…
Est. 95,000 GBP
About the job Help shape the first moments of life for the next generation of AI processors. You'll lead the bring-up and characterisation of Graphcore's silicon devices and system platforms. Your work will ensure produc…
Est. 80,000 GBP
About the job Build the test systems that help deliver AI hardware at production scale. As a Staff Hardware System Test Engineer, you'll develop manufacturing test solutions for complex compute systems. Your work will he…
Est. 65,000 GBP
About the job Build the manufacturing test solutions that help every Graphcore system perform as designed. This role exists to ensure our hardware is tested thoroughly from product introduction through to volume manufact…
Est. 65,000 GBP
About the job Turn product intent into the technical work that moves AI compute forward. As a Technical Product Owner, you will help component-based software teams deliver clear, prioritised and meaningful outcomes. You…
Est. 86,350 GBP
Senior: GBP 73,500 - 99,500 Staff: GBP 97,300 - 131,700 Subject to alignment to the responsibilities and duties of the role - we currently have multiple positions available at both Senior and Staff level. About the job B…
Est. 120,000 GBP
About the job Build the cloud platform that powers Graphcore’s next generation AI systems. As a Principal Cloud Engineer, you will help develop, deploy and operate cloud services across Graphcore’s AI infrastructure. You…
Est. 86,350 GBP
Senior: GBP 73,500 - 99,500 Staff: GBP 97,300 - 131,700 Subject to alignment to the responsibilities and duties of the role - we currently have multiple positions available at both Senior and Staff level. About the job B…
Est. 86,350 GBP
Senior: GBP 73,500 - 99,500 Staff: GBP 97,300 - 131,700 Subject to alignment to the responsibilities and duties of the role - we currently have multiple positions available at both Senior and Staff level. About the job B…
Est. 80,000 GBP
About the job Build the cloud platform that turns Graphcore AI systems into usable, scalable services. Graphcore needs a Bristol-based Staff Cloud Engineer to develop and deploy private and public cloud services for AI c…
Est. 80,000 GBP
About the job Build the cloud platform that puts Graphcore AI systems into real use. As a Senior Cloud Engineer, you will develop and operate private cloud services for Graphcore teams and users. You will turn product an…
Est. 85,000 GBP
About Graphcore 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…
Est. 140,000 USD
About the job Lead the team that proves our AI silicon performs reliably before it reaches customers. You will build and lead Graphcore's characterisation capability for next generation silicon and system platforms. Your…
About the job Help shape how AI hardware reaches production at scale. As a Staff Hardware System Test Engineer, you will build manufacturing test solutions for advanced AI systems. Your work will help ensure every produc…
Est. 80,000 GBP
About Graphcore 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…
Est. 45,000 GBP
About the job Turn complex engineering changes into controlled, traceable actions that keep advanced AI hardware moving forward. As an Engineering Change Analyst, you’ll coordinate product and manufacturing changes acros…
Est. 175,000 USD
About us 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…
Est. 65,000 GBP
About the job Build the systems that prove the next generation of AI hardware works at scale. You will validate and debug Graphcore’s silicon platforms before they power advanced AI workloads worldwide. Your work will di…
Est. 80,000 GBP
About the job Own the firmware that brings Graphcore hardware and silicon to life. As a Staff Firmware Engineer, you will design, develop and deliver firmware for Graphcore’s next generation products. Your work will sit…
Est. 65,000 GBP
About the job Debug the high-speed interfaces that decide whether advanced AI systems perform in the real world. As a Validation and Debug Engineer, you will help prove Graphcore silicon and platforms work across real pr…
Est. 65,000 GBP
About the job Shape how Graphcore products move from hardware design into reliable volume manufacturing. As Staff Manufacturing Product Engineer, you will support NPI development and volume production across complex hard…
Est. 120,000 GBP
About the job Turn complex manufacturing data into higher yield, stronger quality and better products at scale. As Principal Product Engineer, you will help Graphcore manufacture advanced products with optimised yield, q…