Senior AI Performance Engineer

Graphcore
Austin, TX
On-siteCareer-pivot friendly

Who this role is best for

Geared toward engineers comfortable with optimizing AI workloads across distributed systems and communication libraries, with a strong emphasis on collaboration across hardware, software, and networking domains.

Best fit for

  • Candidates with experience in AI performance optimization and distributed communication software
    — “analyze and optimize AI training and inference workloads across large-scale distributed systems”
  • Individuals who can lead technical investigations and validate performance improvements
    — “own complex performance investigations from problem definition through validation”
  • Engineers with a track record of working across multiple layers of the computing stack
    — “works across the computing stack to understand real system behavior”

Things to consider

  • The role is based in Austin, Texas, which may require relocation or local presence
    — “It is based in Austin, Texas.”
  • Candidates must be prepared to coordinate improvements across teams and drive technical decisions
    — “coordinate across teams and validate impact with reliable performance data”

How to stand out

  • Highlight experience with profiling and optimizing AI workloads for large-scale systems
    — “Strong experience profiling and optimizing AI, machine learning or high-performance computing workloads”
  • Showcase skills in distributed communication libraries like MPI, NCCL, UCX, or libfabric
    — “Experience with distributed systems and communication libraries such as MPI, NCCL, UCX and RDMA”
  • Emphasize your ability to design and execute performance benchmarks and models
    — “design benchmarks, investigate system bottlenecks and optimize distributed communication software”
  • Demonstrate understanding of compute, memory, and communication behavior in large systems
    — “Deep understanding of compute, memory and communication behavior in large-scale systems”
  • Demonstrate leadership in driving technical improvements and cross-team collaboration
    — “Ability to own complex technical work and coordinate improvements across teams”
Pace · SteadyCollaboration · HighAutonomy · MediumDecision Impact · Team

Derived from job-description analysis by Serendipath's career intelligence engine.

What success looks like

  • optimized AI training and inference workloads
  • designed benchmarks for system performance
Typical background
strong experience profiling and optimizing AI, machine learning or high-performance computing workloads

Skills & requirements

Required

AI Performance OptimizationDistributed SystemsProfiling And OptimizationCommunication LibrariesC++ And PythonSystem Architecture

Preferred

MlperfAccelerated ArchitecturesML FrameworksHigh-performance Interconnects

Stack & domain

AIMachine LearningHigh-performance ComputingDistributed SystemsMPINCCLUCXRDMAC++PythonMlperfAccelerated ArchitecturesML FrameworksHigh-performance InterconnectsLeadershipCommunicationProblem-solvingTeamworkPerformance EngineeringSystem Architecture

About the role

Original posting from Graphcore via Greenhouse

About the Job

Turn complex AI workload signals into faster, more efficient systems at data center scale.

As a Senior AI Performance Engineer, you will analyze and optimize AI training and inference workloads across large-scale distributed systems. Your work will connect compute, memory, communication and software behavior.

You will own complex performance investigations from problem definition through validation. You will turn profiling, benchmarking and modeling results into improvements engineers can build from.

You will design benchmarks, investigate system bottlenecks and optimize distributed communication software across technologies such as MPI, NCCL, UCX and RDMA. Your work will help Graphcore deliver efficient, reliable AI systems for large-scale deployment.

This role gives you rare scope across hardware, software, networking and system architecture. It is based in Austin, Texas.

The team and culture

The System Engineering Performance team architects, evaluates and optimizes high-performance infrastructure for large-scale data center deployments. The team works across the computing stack to understand real system behavior.

Work moves through evidence, ownership and clear technical judgment. You will define investigations, coordinate across teams and validate impact with reliable performance data.

Decisions are shaped through benchmarks, models, simulations and practical engineering tradeoffs. The team values engineers who think big, act fast, take responsibility, speak up and lead beyond their own area.

What we’re looking 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.

· Ability to own complex technical work and coordinate improvements across teams.

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

While we have outlined a set of requirements, we value transferable skills and diverse experiences. We also welcome engineers returning to the profession after a career break, including through returnship routes.

Benefits

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. 

Graphcore is an equal opportunity employer. We consider qualified applicants without regard to race, color, religion, creed, sex, pregnancy, sexual orientation, gender identity or expression, national origin, ancestry, age, disability, genetic information, veteran status, or any other status protected by applicable law. 

Graphcore is committed to an inclusive and accessible hiring process. If you need a reasonable accommodation to participate in the application or interview process, please let the recruiting team know. 

Personal information submitted during the recruiting process will be handled in accordance with Graphcore's applicable candidate privacy notices.

Join the Team at Graphcore

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.

Ready to improve AI performance across compute, memory, communication and software? Apply now to join Graphcore in Austin

Source: Graphcore careers (Greenhouse)

Similar roles