Graphcore

Graphcore

Staff AI Performance Engineer

Austin, Texas, United States · Staff+

Sponsorship not specifiedDetected 6 days ago
PythonC++Distributed SystemsMachine LearningTensorFlowPyTorchElectrical EngineeringResearchCommunication

About the role

  • Graphcore's AI/ML training and inference infrastructure is rapidly scaling to meet the growing demands of AI workloads across mobile, edge, and datacenter environments.
  • This role focuses on optimizing performance across ARM-based architectures and large-scale distributed systems, ensuring efficiency, scalability, and reliability across the full hardware-software stack.

Responsibilities

  • Collaborate across hardware and software teams to optimize large-scale AI workloads
  • Optimize communication stacks including MPI, NCCL, UCX, RDMA, and networking fabrics
  • Profile and optimize AI workloads, focusing on performance bottlenecks
  • Develop high-quality, ARM-compatible code and documentation
  • The team works across hardware, software, networking, and system architecture to deliver cutting-edge AI solutions and ensure optimal system performance at scale.

Requirements

  • Experience with distributed systems and communication libraries (MPI, NCCL, UCX, libfabric)
  • Experience profiling and optimizing HPC or AI/ML workloads
  • Familiarity with ML benchmarks such as MLPerf
  • Experience with GPUs or accelerated computing architectures

Compensation

  • In addition to a competitive salary, Graphcore offers flexible working and a comprehensive benefits package designed to support your health, wellbeing and financial future.

Company info

  • 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 enjoys a culture of continuous learning and constant innovation.
  • 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 enjoys a culture of continuous learning and constant innovation.
  • The System Engineering Performance team architects and optimizes high-performance infrastructure for large-scale datacenter deployments.
  • and Duties
  • Analyze ML models' compute and memory requirements using roofline analysis and simulations
  • Benchmark, monitor, and troubleshoot system performance across distributed systems
  • Candidate Profile
  • BS/MS in Computer Science, Electrical Engineering, or related field
  • Strong programming skills in C++ and Python
  • Knowledge of HPC networking and interconnect technologies (InfiniBand, RoCE)
  • Familiarity with ML frameworks such as PyTorch or TensorFlow
  • Understanding of ARM architectures and toolchains
  • Strong debugging, profiling, and performance optimization skills
  • In addition to a competitive salary, Graphcore offers flexible working and a comprehensive benefits package designed to support your health, wellbeing and financial future.
  • Our benefits include medical, dental and vision coverage, Flexible Spending Accounts (FSAs), Health Savings Accounts (HSAs), disability and life insurance, a 401(k) retirement plan, commuter benefits, wellness services and an Employee Assistance Programme (EAP).
Apply directly at Graphcore →Create a free account for alerts like thisView Graphcore immigration profile

This listing is sourced directly from Graphcore's careers page and normalized into a canonical job model.