Software Engineer - Triton
Job description
Software engineering role in Graphcore's Frameworks team, specifically the Triton team, building and optimising framework support so ML frameworks such as Triton and PyTorch work with Graphcore accelerators.
At a glance
- Work mode
- Remote · United Kingdom
What you'll do
- Design, implement, optimise and support critical software connecting Graphcore accelerators with ML frameworks including Triton and PyTorch
- Deliver new features in the Triton team
- Review code and write technical documentation
- Work with upstream open-source communities
- Help coordinate open-ended technical work across teams
- Improve performance, quality and usability across a complex software stack
- May contribute across other framework teams
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
Your team
Frameworks team (Triton team)
The Frameworks team connects Graphcore hardware with the tools ML engineers and researchers rely on, working across PyTorch, Triton, JAX and TensorFlow with a focus on usability and performance.
Benefits
Workplace experience: Enjoy thoughtfully designed office spaces for collaboration, with free food and an on-site barista to support your day
Flexible benefits: Tailor your benefits package with a choice of additional options, including private medical insurance and dental cover
- Flexible working: Balance your work and personal life with greater flexibility
- Generous leave: Take time to rest, recharge and enjoy life outside of work
- Retirement planning support: Up to 5% matched pension
- Phantom equity: Share in Graphcore's success
- Peace of mind protection: Income protection and life assurance to provide financial security for you and your loved ones
- Optional benefits: Dental cover, health cash plan, private medical insurance, cycle to work scheme, give as you earn
Apply
About Graphcore
One of the world's leading innovators in Artificial Intelligence compute
Graphcore is developing hardware, software and systems infrastructure that will unlock the next generation of AI breakthroughs and power the widespread adoption of AI solutions across every industry. As part of the SoftBank Group, Graphcore is a member of an elite family of companies responsible for some of the world's most transformative technologies.
Mission
To enable Artificial Super Intelligence and ensure its benefits are accessible to everyone
Key facts
- Artificial Intelligence compute
- Part of the SoftBank Group
- Teams include AI research specialists, silicon designers, software engineers and systems architects
What they value
- Inclusive environment where everyone can do their best work
- Evidence-based decisions shaped by code quality and business outcomes
- Clear ownership and trust in engineers to speak up and solve hard problems