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 role offers 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, focusing 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 work across Graphcore and with open-source communities to deliver software 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. Benefits · Flexible working: Balance your work
Salary not disclosed
Check market pay for comparable PyTorch Engineer roles before applying.
Role overview
Job 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 t
…What they are looking for
Skills & requirements
Department · Frameworks
Hiring company
Graphcore
Graphcore has built a new type of processor for machine intelligence to accelerate machine learning and AI applications for a world of intelligent machines
Related opportunities
Similar active roles
Relevant, currently active roles for this market.
🔔 Get job alerts
New PyTorch Engineer jobs in Bristol, United Kingdom, straight to your inbox.
No spam · Unsubscribe anytime