Senior AI Performance Engineer - Data-Center Scale

Graphcore; Milpitas, Northern (CA, KY); Hybrid; USD 210,000 - 260,000; Full time

Benefits offered by this job

Medical, dental, and vision coverage

Flexible working hours

Hybrid working arrangements

Professional-development resources

Job summary

Graphcore in Milpitas, California, seeks a Staff AI Performance Engineer to analyze and optimize AI training and inference workloads across distributed systems, connecting compute, memory, networking and software behavior.

You will lead cross-team performance workstreams, identify bottlenecks, validate improvements with benchmarks, and develop C++ and Python tools to enable faster, more scalable AI systems.

Qualifications

• Experience profiling and optimizing AI/ML or HPC workloads.
• Strong C++ and Python skills with performance-sensitive software.
• Deep understanding of compute, memory, and communication in large-scale systems.

Responsibilities

• Lead performance workstreams across hardware and software.
• Identify bottlenecks and validate improvements with reliable performance data.
• Develop C++ and Python tools for analysis and optimization.

Skills

C++

Python

Profiling & optimization

Distributed systems

Tools

MPI

NCCL

UCX

libfabric

Job description

Graphcore in Milpitas, California, seeks a Staff AI Performance Engineer to analyze and optimize AI training and inference workloads across distributed systems, connecting compute, memory, networking and software behavior.

You will lead cross-team performance workstreams, identify bottlenecks, validate improvements with benchmarks, and develop C++ and Python tools to enable faster, more scalable AI systems.