PyTorch Engineer

Salary Range: 260,400 - 352,200

Subject to alignment to the responsibilities and duties of the role

About the job

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 culture

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

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 shape how developers use PyTorch on advanced AI hardware, we’d love to hear from you. Apply now to be part of the journey.