Software Engineer - Triton

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. Reporting to a Team Lead in Frameworks, you will design, implement, optimise and support critical software.

Your work will shape how researchers and engineers use Graphcore hardware for demanding machine learning workloads. You will improve performance, quality and usability across a complex software stack.

You will join our Triton team, delivering new features, reviewing code, writing technical documentation and working with upstream communities. You will also help coordinate open-ended technical work across teams.

This is a role for engineers who want deep technical ownership and visible impact. As the ML software landscape evolves, you will keep learning and may contribute across other framework teams.

The team and culture

The Frameworks team connects Graphcore hardware with the tools ML engineers and researchers rely on. We work across PyTorch, Triton, JAX and TensorFlow, with a strong focus on usability and performance.

Work happens through agile practices, close technical discussion and clear ownership. Engineers are trusted to speak up, solve hard problems and move work forward with pace.

Decisions are shaped by evidence, code quality and business outcomes. You will collaborate across Graphcore and with open-source communities to deliver software that developers can trust.

What we’re looking for

Strong software engineering skills, with the judgement to manage quality, complexity and technical debt.

Experience developing in Python and C++.

Experience in compiler development.

Ability to manage complex technical tasks with cross-team dependencies.

Clear communication skills, with a proactive approach to collaboration and technical leadership.

Interest in AI, ML frameworks, performance optimisation or computationally intensive engineering.

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.