Software Engineer (Triton)
Posted
Summary
Software Engineer on the Triton team building framework support (PyTorch, Triton, JAX, TensorFlow) for Graphcore's AI accelerators, focusing on compiler development, performance optimization, and open-source collaboration using Python and C++.
- 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 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
Benefits
- Work-Life Balance: We believe you should have a good work-life balance. Everyone’s situation is different, so we offer flexible working arrangements and start and finish times, allowing you to choose when to work from home or come into the office
- Healthcare: We offer Private Medical Insurance (PMI) to our employees to help them take care of their health and that of their loved ones
- Pension: We believe that our employees deserve the best when it comes to planning for their future
- Income Protection: Graphcore understands that unforeseen circumstances can happen, and we want to ensure our employees have the necessary financial security in place
- Progression: We prioritise learning and development as an integral part of our culture at Graphcore. Our Talent Development team is dedicated to equipping all our employees with the necessary tools, resources, and support to enhance their skills and reach their full potential in their careers
- Wellbeing: We want to ensure that our employees have the resources and support they need to prioritise their health and wellbeing
- Interest in AI, ML frameworks, performance optimisation or computationally intensive engineering
- Strong software engineering skills, with the judgement to manage quality, complexity and technical debt
- Clear communication skills, with a proactive approach to collaboration and technical leadership
- Ability to manage complex technical tasks with cross-team dependencies
- Experience developing in Python and C++
- Experience in compiler development
