Staff AI Performance Engineer

About the Company

Graphcore is one of the world’s leading innovators in Artificial Intelligence compute. As part of the SoftBank Group, Graphcore is developing hardware, software and systems infrastructure that will unlock the next generation of AI breakthroughs.

Responsibilities

• Analyze ML models’ compute and memory requirements using roofline analysis and simulations
• Collaborate across hardware and software teams to optimize large-scale AI workloads
• Benchmark, monitor, and troubleshoot system performance across distributed systems
• Optimize communication stacks including MPI, NCCL, UCX, RDMA, and networking fabrics
• Profile and optimize AI workloads, focusing on performance bottlenecks
• Develop high-quality, ARM-compatible code and documentation

Requirements

• BS/MS in Computer Science, Electrical Engineering, or related field
• Experience with distributed systems and communication libraries (MPI, NCCL, UCX, libfabric)
• Strong programming skills in C++ and Python
• Experience profiling and optimizing HPC or AI/ML workloads
• Familiarity with ML benchmarks such as MLPerf

Preferred Qualifications

• Experience with GPUs or accelerated computing architectures
• Knowledge of HPC networking and interconnect technologies (InfiniBand, RoCE)
• Familiarity with ML frameworks such as PyTorch or TensorFlow
• Understanding of ARM architectures and toolchains
• Strong debugging, profiling, and performance optimization skills

Benefits

• Medical, dental and vision coverage
• Flexible Spending Accounts (FSAs) and Health Savings Accounts (HSAs)
• Disability and life insurance
• 401(k) retirement plan
• Commuter benefits and wellness services