AI Software Engineer

Graphcore is one of the world’s leading innovators in Artificial Intelligence compute. It 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. Together, they share a bold vision: to enable Artificial Super Intelligence and ensure its benefits are accessible to everyone.

A melting pot of AI research specialists, silicon designers, software engineers and systems architects, Graphcore enjoys a culture of continuous learning and constant innovation. Reporting to a Team Lead in the frameworks team you will play a pivotal role in designing, implementing, optimising, maintaining and supporting the software required to ensure the machine learning accelerators that Graphcore develop enjoy first-class support in state-of-the-art machine learning frameworks such as PyTorch and Triton.

This role sees you joining our Triton team, where you will be part of a SCRUM team working on delivering new features, optimising performance, reviewing code changes, writing technical documentation, working with upstream communities, maintaining the code base and supporting users. You help the team coordinate and deliver complex, open-ended technical tasks. You will develop deep expertise in the Triton project and will contribute to the team’s technical direction and processes.

You understand the importance of managing code quality and code complexity and balancing this against the need to deliver business outcomes. The Frameworks team ensures Graphcore hardware works seamlessly with the tools that ML engineers and researchers love – Pytorch, Triton, Jax and TensorFlow. All our software teams follow agile working practices, and we care deeply about both ease-of-use as well as performance. We work closely 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.

• 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.
• Resolving regressions, performance issues and software defects.
• Coaching and mentoring other team members.

Demonstrable strong software engineering skills. Demonstrable experience of Python/C++ development. Demonstrable knowledge of AI/ML. Experience in profiling / optimising high performance code. Experience of writing ML kernels. Experience of using and/or development of ML Frameworks. Experience in a computationally intensive engineering field.

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’re committed to building an inclusive work environment that makes Graphcore a great home for everyone. We can provide a flexible approach to interview and encourage you to chat to us if you require any reasonable adjustments. Unfortunately at this time, we are unable to provide visa sponsorship or support for visa applications.