Software Engineer - Pytorch
GraphcoreBristol, England, United Kingdom • Cambridge, England, United Kingdom
Posted October 3, 2025
Full Time
About the Company
Graphcore is at the forefront of artificial intelligence compute with its Intelligence Processing Unit (IPU), a uniquely architected processor designed to accelerate machine learning models by orders of magnitude. The company's 3D Wafer-on-Wafer Bow IPU systems enable groundbreaking AI research and innovation across various applications, including language processing, computer vision, and graph neural networks. Graphcore aims to become the global standard for AI compute, transforming industries from healthcare to autonomous vehicles through its advanced technology.
Job Description
Graphcore is seeking a Software Engineer to contribute to their PyTorch team. This role involves designing, implementing, optimizing, and maintaining software that allows the development of AI hardware. The role offers opportunities to collaborate with other engineers, manage code quality, and contribute to technical direction. The team works with frameworks like PyTorch and Triton, and focuses on performance and user experience.
Requirements
- Demonstrable strong software engineering skills
- Experience of Python development
- Experience of developing performant C++ applications
- Experience of writing ML kernels
- Experience of using and/or development of ML Frameworks
- Experience in a computationally intensive engineering field
- Understanding of computing architectures
- Knowledge of AI/ML
- Understanding of complex code bases
- Experience in profiling / optimising high performance code
Benefits
- Competitive salary
- Flexible working
- Private medical insurance
- Health cash plan
- Dental plan
- Pension
- Life assurance
- Income protection
- Employee assistance programme