ML Frameworks Engineer - Triton & PyTorch

AI Startups UK; West of England; Hybrid; GBP 90,000 - 130,000; Full time

Benefits offered by this job

Flexible working

Generous leave

Pension matching

Phantom equity

Free food & on-site barista

Income protection & life assurance

Private medical insurance

Cycle to work

Dental cover

Job summary

Graphcore is looking for an ambitious software engineer to join the Frameworks team and own parts of the ML software stack. You will help Graphcore accelerators work seamlessly with modern ML frameworks, including Triton and PyTorch, by designing, implementing, optimising and supporting critical software.

Reporting to a Team Lead, you will deliver new features, review code, write technical documentation and coordinate work across teams.

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 ML software.
• Deliver features for Graphcore frameworks and accelerators.
• Review code and write technical documentation.
• Coordinate open-ended technical work across teams.

Skills

Software engineering

Python

C++

Compiler development

Cross-team coordination

Communication

AI/ML frameworks

Job description

Graphcore is looking for an ambitious software engineer to join the Frameworks team and own parts of the ML software stack. You will help Graphcore accelerators work seamlessly with modern ML frameworks, including Triton and PyTorch, by designing, implementing, optimising and supporting critical software.

Reporting to a Team Lead, you will deliver new features, review code, write technical documentation and coordinate work across teams.