Senior Software Engineer - AI Compute Libraries & Performance
Core
Build high-performance AI compute libraries (linear algebra, tensor ops) for next-generation AI hardware, ensuring numerical correctness and low-level performance optimisation.
Role type
Senior IC software engineer (AI compute kernels)
Builds
High-performance compute libraries for AI hardware
Domain
Semiconductor / AI Infrastructure
Deliverable
production ML models
Required skills
C++, Python, Linux profiling, processor architecture knowledge, threading, cache locality optimization, memory layout optimization, kernel launch efficiency, microbenchmarking, regression testing, numerical validation
Preferred skills
Agile methodologies, mentoring
Technologies
C++, Python, Linux
Responsibilities
Design and implement kernels for linear algebra and tensor ops (GEMM, batched GEMM, convolutions, reductions, elementwise and fused operations); Own performance and correctness by adding microbenchmarks, regression tests, and numerics validation; Profile and optimise across next generation of AI hardware (threading, cache locality, memory layout, kernel launch efficiency); Debug issues and improve product quality; Mentor colleagues within the team
Seniority
Senior, hands-on IC