PyTorch Engineer

Reposted 11 Hours Ago
Be an Early Applicant
Bristol, England, GBR
Hybrid
Senior level
Artificial Intelligence • Semiconductor
Joining Graphcore gives you a seat at the top-table, shaping the future of Artificial Intelligence.
The Role
Design, implement, optimise and maintain PyTorch framework support for Graphcore accelerators. Deliver features, tests, performance improvements, documentation and code reviews; collaborate across teams and upstream open-source projects; mentor colleagues and contribute to technical direction.
Summary Generated by Built In
About the job

Make PyTorch feel native, powerful and production-ready on Graphcore AI compute.

You will help build the software layer that connects Graphcore accelerators with state-of-the-art machine learning frameworks. Your work will shape how ML engineers and researchers use our hardware in real workloads.

You will design features, improve performance, maintain core code, and support users working at the frontier of AI. You will work deeply in PyTorch, with scope to contribute across other frameworks as the landscape evolves.

This is a role for engineers who enjoy hard technical problems and open-ended ownership. You will influence architecture, quality and direction while helping Graphcore unlock the next generation of AI breakthroughs.

The team and culture

The Frameworks team makes Graphcore hardware work seamlessly with the tools ML engineers already use, including PyTorch, Triton, JAX and TensorFlow. Work happens through agile practices, clear technical debate and close links with hardware, software and open-source teams.

You will join a Scrum team that owns features from design through delivery, performance, documentation and support. Engineers are expected to take responsibility, speak up early and make decisions grounded in evidence.

The team values pace without losing quality. You will be trusted to coordinate complex work, manage trade-offs and help others grow through reviews, mentoring and shared learning.

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 how high-performance software works.

Benefits

· Unlimited annual leave
· Up to 5% matched pension
· Phantom equity – share in Graphcore’s success
· True flexibility in how and where you work
· Office spaces designed for collaboration
· Free food and an on-site barista
· Health cash plan
· Income protection
· Life assurance
· Along with other benefits you can choose from (private medical insurance, dental plan etc)

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 make advanced AI compute easier, faster and more useful for researchers and engineers, 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: writing ML kernels, compiler development, development/usage of ML frameworks, or experience in computationally intensive engineering
  • Right to work in the UK (no visa sponsorship available)
  • 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

Monika
Dionysia
Dave

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

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Bristol
903 Employees
Year Founded: 2016

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

Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
Gallery
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.

Typical time on-site: 3 days a week
HQHeadquarters
Austin Office
Bengaluru Office
Cambridge Office
Gdańsk Office
Hsinchu Office
London Office
Learn more

Similar Jobs

Graphcore Logo Graphcore

Software Infrastructure Kubernetes Engineer

Artificial Intelligence • Semiconductor
Hybrid
Bristol, England, GBR
903 Employees

Graphcore Logo Graphcore

Software Engineer

Artificial Intelligence • Semiconductor
Hybrid
Bristol, England, GBR
903 Employees

Graphcore Logo Graphcore

Director Of Product Engineering

Artificial Intelligence • Semiconductor
Hybrid
Bristol, England, GBR
903 Employees

Graphcore Logo Graphcore

Debug Validation Engineer — Multiple Levels

Artificial Intelligence • Semiconductor
Hybrid
Bristol, England, GBR
903 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account