Software Engineer - Triton Bristol, UK; Cambridge, UK

Overview

Graphcore is a world-leading innovator in Artificial Intelligence compute. It develops hardware, software and systems infrastructure to unlock the next generation of AI breakthroughs and power widespread adoption of AI solutions across industries. As part of the SoftBank Group, Graphcore is a member of an elite family of companies responsible for transformative technologies, with a bold vision to enable Artificial Super Intelligence and ensure its benefits are accessible to everyone. Graphcore’s teams bring a diverse range of skills and perspectives, including AI research specialists, silicon designers, software engineers and systems architects, in a culture of continuous learning and innovation.

The Role

Reporting to a Team Lead in the frameworks team, you will design, implement, optimise, maintain and support the software required to ensure the machine learning accelerators Graphcore develops enjoy first-class support in state-of-the-art frameworks such as PyTorch and Triton (note: Triton refers to the Triton language/compiler and not the inference framework).

You will join our Triton team as part of a SCRUM team, delivering new features, optimising performance, reviewing code changes, writing technical documentation, engaging with upstream communities, maintaining the code base and supporting users.

In this role you will collaborate closely with other engineers within the Triton team and across engineering teams. You will coordinate and deliver complex, open-ended technical tasks, be proactive and an excellent communicator, develop deep expertise in the Triton project and contribute to the team’s technical direction and processes. You will shape code quality and complexity while balancing delivery of business outcomes. Over time you may be asked to work on teams beyond Triton as the ML software landscape moves quickly.

The Team

The Frameworks team ensures Graphcore hardware works seamlessly with the tools ML engineers and researchers love – PyTorch, Triton, Jax and TensorFlow. We are a diverse, collaborative team that values openness, learning, and a balance of ease-of-use and performance. We work with other Graphcore teams and leading open-source communities.

Joining us means contributing to cutting-edge work in the machine learning industry and enabling ML engineers and researchers to unlock the potential of Graphcore’s hardware.

Responsibilities and Duties

• Design and implement new features; maintain and support existing features.
• Develop and maintain unit tests, component tests and integration tests.
• Optimize the software stack for performance for our users.
• Manage complex technical tasks with cross-team dependencies.
• Manage code quality, code complexity and technical debt.
• Provide technical leadership and contribute to the team’s direction and processes.
• Contribute to documentation, including user manuals and tutorials.
• Conduct code reviews.
• Resolve regressions, performance issues and software defects.
• Coach and mentor other team members.
• Foster a collaborative team culture and continuous improvement of ways of working.
• Collaborate with upstream open-source development teams.

Candidate Profile

• Strong software engineering skills.
• Experience in Python/C++ development.
• Experience in compiler development.
• Experience with MLIR (or SSA-based IRs) is highly desirable.
• Knowledge of AI/ML and understanding of computing architectures.
• Experience maintaining and supporting a complex code base.
• Experience profiling/optimising high-performance code.
• Experience writing ML kernels and using or developing ML Frameworks.
• Experience in a computationally intensive engineering field.

In addition to a competitive salary, Graphcore offers flexible working, generous leave, private medical insurance and health cash plan, dental plan, pension (matched up to 5%), life assurance and income protection. We provide parental leave, an employee assistance programme, healthy food and beverages at our Bristol office and a barista bar. We are committed to an inclusive work environment and offer an equal opportunity process with flexible interview adjustments as needed.

Applicants for this position must hold the right to work in the UK. We are unable to provide visa sponsorship or support for visa applications at this time.

We welcome applications from diverse backgrounds and experiences and strive to create an inclusive culture.

EEO and Recruitment Notes

We take pride in our commitment to inclusion and diversity in the workplace. As part of our recruitment process, we may collect confidential diversity data that will be anonymised and used for statistical purposes to improve our hiring process. Responses are confidential and do not impact your application.