Senior Software Engineer - AI Compute Libraries & Performance

1 week agoSeniorSalary: 260K - 352KGdańsk, Pomeranian Voivodeship, Poland Engineering Jobs by Graphcore

Skills

Bf16 Blas C#Dnn Fp16 Fp32 Fp8 Kernel Linux Memory Hierarchy Microbenchmarking Numerical Stability Processor Architecture Profiling Python Regression Testing Threading Vectorisation

Candidate Availability

Gdańsk · Required Poland · Required Pomeranian Voivodeship · Required

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You build high-performance AI compute libraries for next-generation hardware, implement kernels for linear algebra and tensor operations, own performance and numerical correctness through benchmarks and tests, profile and optimize threading and memory behaviour, debug and resolve issues, and mentor others.

Requirements

Excellent C++ programming skills
Excellent Python scripting skills
Understanding of processor architectures
Linux profiling
Algorithmic performance
Vectorisation
Memory hierarchy
Threading
Hands-on experience with BLAS or DNN stacks
Ability to read or extend kernels
Understanding of numerical stability
Understanding of FP32, FP16, BF16 and FP8 trade-offs
Written and verbal communication skills

Responsibilities

Build high-performance AI compute libraries
Design and implement linear algebra and tensor kernels
Implement GEMM, convolution and reduction operations
Own performance and correctness through microbenchmarks
Maintain regression tests
Perform numerics validation
Profile and optimize threading, cache locality, memory layout and kernel launch efficiency
Debug and resolve bugs
Mentor others
Support Agile ways of working

Benefits

Flexible working
Comprehensive healthcare and dental cover
Phantom equity
Fully funded gym card
Fitness classes
Death and disability insurance
Retirement savings with a 4% employer contribution
Employee Assistance Programme