Software Engineer - ML Kernels & Runtime
As a Software Engineer at Graphcore, you will be responsible for developing and maintaining kernels for linear algebra operations on our next-generation AI hardware. This role is integral to our Kernel Engineering team, which focuses on delivering high-performance compute libraries to maximize the capabilities of our AI processors. Graphcore is a leading innovator in AI compute, dedicated to building the complete AI stack from silicon to software.
In this position, your primary responsibilities will include implementing kernels for linear algebra and tensor operations such as GEMM, convolutions, and reductions using C++. You will profile and optimize these kernels to enhance performance across our AI hardware, addressing aspects like threading, cache locality, and memory layout. Additionally, you will contribute to performance and correctness by adding microbenchmarks, regression tests, and numerical validation, as well as debugging issues and improving product quality.
The ideal candidate will possess excellent programming and scripting skills in C++ and Python, along with a strong understanding of processor architectures. You should thrive on solving challenging problems creatively, think beyond specific software components, and have excellent written and oral communication skills. A high sense of teamwork and a commitment to producing quality work are essential.
Graphcore offers a competitive salary, flexible working arrangements, a generous annual leave policy, private medical insurance, a dental plan, and a pension matched up to 5%. Additional benefits include life assurance, income protection, a generous parental leave policy, and an employee assistance program covering health and mental wellbeing. Our central Bristol office provides a range of healthy food and snacks, including a barista bar, fostering an inclusive and supportive work environment.