2026 Graduate Software Engineer - PyTorch

The Frameworks team ensures Graphcore hardware works seamlessly with the tools that ML engineers and researchers love – Pytorch, Triton and in future, Jax and TensorFlow. We’re a talented and diverse team of engineers and we foster a culture of collaboration, openness and learning. All our software teams follow agile working practices, and we care deeply about both ease-of-use as well as performance. We work closely work with other Graphcore teams as well as leading open-source communities. By joining us, you’ll join our exciting journey on the cutting edge of the machine learning industry. Your contributions will make a real difference – enabling machine learning engineers and researchers to unlock the full potential of Graphcore’s hardware. Responsibilities and Duties

• Supporting the design and implementation of new features & maintaining and supporting existing features. - Developing and maintaining unit tests, component tests and integration tests. - Optimising the software stack to make it more performant for our users. - Managing code quality, code complexity and technical debt. - Contributing to documentation, including user manuals and tutorials. - Conducting code reviews. - Resolving regressions, performance issues and software defects. - Contribute to continuous improvements to improve ways of working. - Working with upstream open-source development teams. Candidate Profile

Essential:

• Bachelor or Master's in Computer Science, Maths, Machine Learning, Data Science, or related field

• Experience of Python development. - At least one of:

• Experience of compiler development. - Experience of using and/or development of ML Frameworks. - Experience in a computationally intensive engineering field. Desirable:

• Demonstrable knowledge of AI/ML. - Understanding of computing architectures. - Experience of maintaining and supporting a complex code base. - Experience in profiling / optimising high performance code. - Experience of writing ML kernels. Benefits

In addition to a competitive salary, Graphcore offers flexible working, a generous annual leave policy, private medical insurance and health cash plan, a dental plan, pension (matched up to 5%), life assurance and income protection. We have a generous parental leave policy and an employee assistance programme (which includes health, mental wellbeing, and bereavement support). We offer a range of healthy food and snacks at our central Bristol office and have our own barista bar! We welcome people of different backgrounds and experiences; we’re committed to building an inclusive work environment that makes Graphcore a great home for everyone. We offer an equal opportunity process and understand that there are visible and invisible differences in all of us. We can provide a flexible approach to interview and encourage you to chat to us if you require any reasonable adjustments.