About this role

Graphcore is seeking a Staff AI Performance Engineer to optimize AI/ML training and inference infrastructure across ARM-based architectures and distributed systems. This role involves analyzing model requirements, collaborating with hardware and software teams, and profiling/optimizing workloads for efficiency and scalability.

What we are looking for

6

Optimize AI/ML training and inference infrastructure
Analyze ML models' compute and memory requirements
Collaborate across hardware and software teams
Benchmark, monitor, and troubleshoot system performance
Profile and optimize AI workloads for performance bottlenecks
Develop high-quality, ARM-compatible code

Skills mentioned

2

PythonC++