Drimva

Graphcore

Senior AI Performance Engineer

Austin, Texas, United States · Senior · Full-time · Posted Oct 9, 2026

About the job

Analyzes and optimizes AI training and inference workloads across distributed systems, turning performance investigations into validated system improvements.

What you'll do

  • Profile, benchmark, and model workloads to identify bottlenecks and improve performance.
  • Design benchmarks and optimize distributed communication software.
  • Own complex investigations and coordinate improvements across teams.

What they ask for

  • Strong experience profiling and optimizing AI, machine learning, or high-performance computing workloads.
  • Experience with distributed systems and communication libraries such as MPI, NCCL, UCX, or libfabric.
  • Strong C++ and Python skills, including reliable tools or performance-sensitive software.
  • Deep understanding of compute, memory, and communication behavior in large-scale systems.

Nice to have

  • Familiarity with MLPerf, accelerated architectures, ML frameworks, or high-performance interconnects.

Tools

MPI, NCCL, UCX, libfabric, C++, Python, MLPerf

AI focus

Optimizes AI training and inference performance in large-scale distributed systems.

Model-written summary · the original posting is authoritative

First seen Oct 27, 2026 · Last verified Nov 5, 2026 (UTC)