Software Engineer - ML Kernels & Runtime

Job Summary

As a Software Engineer you will be responsible for development of new and support existing kernels for liner algebra operations on a new generation of AI hardware.

The Team

This is an exciting opportunity to join an expanding team at Graphcore. Kernel Engineering team is responsible for delivering high performance compute library to help customers gain the maximum performance from AI hardware.

Responsibilities and Duties

• Design and implement kernels for linear algebra and tensor ops (GEMM, batched GEMM, convolutions, reductions, elementwise and fused operations) in C++
• Profile and optimise across for next generation of AI hardware - threading, cache locality, memory layout, and kernel launch efficiency.
• Own performance and correctness - add microbenchmarks, regression tests, numerics validation
• Debug issues, resolve bugs and generally improve the quality and functionality ...