Senior AI Performance Engineer for Large-Scale Systems

Graphcore; Austin (TX); Hybrid; USD 190,000 - 260,000; Full time

Benefits offered by this job

Medical, dental, and vision coverage

Mental health resources

Retirement savings benefits

Paid vacation and holidays

Life insurance and disability coverage

Flexible working hours and hybrid work

Job summary

Graphcore in Austin, Texas seeks a Senior AI Performance Engineer to analyze and optimize AI training and inference workloads across large-scale distributed systems. You will own performance investigations from problem definition through validation, turning profiling, benchmarking and modeling results into concrete improvements.

You will design benchmarks and investigate bottlenecks, optimizing distributed communication software across technologies such as MPI, NCCL, UCX and RDMA.

Qualifications

• 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.
• Ability to own complex technical work and coordinate improvements across teams.
• Familiarity with MLPerf, accelerated architectures, ML frameworks or high-performance interconnects.

Responsibilities

• Analyze and optimize AI training and inference workloads across large-scale distributed systems.
• Own complex performance investigations from problem definition through validation.
• Turn profiling, benchmarking and modeling results into actionable improvements for engineers.
• Design benchmarks, investigate system bottlenecks and optimize distributed communication software (MPI/NCCL/UCX).
• Contribute to cross-team collaboration and technical decision-making with data-driven evidence.
• Engage with hardware, software, networking and system architecture considerations to improve AI compute performance.

Skills

AI profiling

Distributed systems

MPI/NCCL/UCX

C++

Python

Performance optimization

Systems architecture

Cross-team collaboration

Job description

Graphcore in Austin, Texas seeks a Senior AI Performance Engineer to analyze and optimize AI training and inference workloads across large-scale distributed systems. You will own performance investigations from problem definition through validation, turning profiling, benchmarking and modeling results into concrete improvements.

You will design benchmarks and investigate bottlenecks, optimizing distributed communication software across technologies such as MPI, NCCL, UCX and RDMA.