Description
Job Overview
As a Software Engineer - Triton at Graphcore, you will play a crucial role in building framework support to help AI developers unlock Graphcore hardware. You will work on making Graphcore accelerators work seamlessly with state-of-the-art ML frameworks, including Triton and PyTorch. Your responsibilities will include designing, implementing, optimizing, and supporting critical software that shapes how researchers and engineers use Graphcore hardware for demanding machine learning workloads.
Responsibilities
- Design, implement, optimize, and support critical software for Graphcore hardware
- Help improve performance, quality, and usability across a complex software stack
- Collaborate with the Triton team to deliver new features, review code, write technical documentation, and work with upstream communities
- Coordinate open-ended technical work across teams
Requirements
- Strong software engineering skills with judgment 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 optimization, or computationally intensive engineering
Benefits
- Flexible working
- Comprehensive healthcare and dental cover
- Phantom equity
- Fitness and wellbeing support
- Financial protection
- Retirement savings
- Employee Assistance Programme (EAP)
Team and Culture
The Frameworks team connects Graphcore hardware with tools ML engineers and researchers rely on, working across PyTorch, Triton, JAX, and TensorFlow with a strong focus on usability and performance.