ML Frameworks Engineer: Triton & PyTorch Integration

Graphcore; West of England; On-site; GBP 90,000 - 130,000; Full time

Benefits offered by this job

Generous leave

Pension matching

Phantom equity

Free food on-site

Life insurance

Private medical insurance

Dental cover

Cycle to work

Flexible benefits

Diversity & inclusion

Job summary

Graphcore is seeking a skilled ML software engineer to build framework support that helps AI developers unlock Graphcore hardware. You will ensure accelerators work seamlessly with ML frameworks like Triton and PyTorch, design and optimise software, and contribute across the framework stack with strong ownership.

The role offers deep technical ownership, opportunities to work with open-source communities, and the chance to influence how researchers and engineers use Graphcore hardware for

Qualifications

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

Responsibilities

• Design, implement, optimise and support critical software.
• Help Graphcore accelerators work seamlessly with state-of-the-art ML frameworks, including Triton and PyTorch.
• Collaborate with upstream communities and coordinate open-ended technical work across teams.
• Deliver new features, review code and write technical documentation.
• Provide deep technical ownership and visible impact within the ML software landscape.

Skills

Python

C++

Software engineering

Compiler development

Cross-team collaboration

Technical leadership

Tools

PyTorch

Triton

Job description

Graphcore is seeking a skilled ML software engineer to build framework support that helps AI developers unlock Graphcore hardware. You will ensure accelerators work seamlessly with ML frameworks like Triton and PyTorch, design and optimise software, and contribute across the framework stack with strong ownership.

The role offers deep technical ownership, opportunities to work with open-source communities, and the chance to influence how researchers and engineers use Graphcore hardware for