Staff AI Performance Engineer

About the Position

Staff AI Performance Engineer

Milpitas, California, United States

---

About the Job

Lead performance work that turns large-scale AI workloads into faster, more efficient systems.

As a Staff AI Performance Engineer, you will analyze and optimize AI training and inference workloads across distributed systems. Your work will connect compute, memory, networking, and software behavior.

You will lead complex performance workstreams across hardware and software. You will find bottlenecks, validate improvements, and turn performance evidence into practical engineering decisions.

You will build benchmarks, profile demanding workloads, and develop C++ and Python tools for analysis and optimization. Your recommendations will help teams improve efficiency, scalability, and reliability.

This role gives you broad technical scope across AI infrastructure at data center scale. It is based in Milpitas, California.

---

Responsibilities

• Analyze and optimize AI training and inference workloads across distributed systems
• Connect compute, memory, networking, and software behavior
• Lead complex performance workstreams across hardware and software
• Find bottlenecks and validate improvements
• Develop C++ and Python tools for analysis and optimization

---

Requirements

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

---

We Offer

Graphcore offers competitive compensation and benefits designed to support employees' health, financial well-being, work-life needs, and professional growth. Benefits and programs for eligible U.S. employees may include:

• Medical, dental, and vision coverage, with options that may extend to eligible dependents
• Mental health, wellness, and employee assistance resources
• Retirement savings benefits and company contributions where applicable
• Paid vacation, sick time, company holidays, and parental or family leave in accordance with applicable plans and policies
• Life insurance and short-term or long-term disability coverage
• Flexible working hours and hybrid working arrangements where compatible with the role and team requirements
• Professional-development resources, learning programs, office amenities, and team-led activities

Benefits vary by work location, employment status, scheduled hours, and plan eligibility and are subject to the terms of the applicable plans and company policies. This overview is not a contract or guarantee of benefits.

---

About the Company

Graphcore is one of the world’s leading innovators in Artificial Intelligence compute. It is developing hardware, software, and systems infrastructure that will unlock the next generation of AI breakthroughs and power the widespread adoption of AI solutions across every industry.

As part of the SoftBank Group, Graphcore is a member of an elite family of companies responsible for some of the world’s most transformative technologies. Together, they share a bold vision: to enable Artificial Super Intelligence and ensure its benefits are accessible to everyone.

Graphcore’s teams are drawn from diverse backgrounds and bring a broad range of skills and perspectives. A melting pot of AI research specialists, silicon designers, software engineers, and systems architects, Graphcore brings together deep expertise to solve complex problems and deliver meaningful progress in AI compute.

---

Join the Team at Graphcore

Ready to lead performance improvements across AI systems at data center scale? Apply now to join Graphcore in Milpitas.