15 hours ago
Responsibilities
- Analyze and optimize AI training and inference workloads across large-scale distributed systems.
- Own performance investigations from problem definition through validation.
- Design benchmarks and investigate compute, memory, communication, and software bottlenecks.
- Optimize distributed communication software using technologies such as MPI, NCCL, UCX, and RDMA.
- Translate profiling, benchmarking, and modeling results into actionable engineering improvements.
- Coordinate improvements across teams and validate impact with reliable performance data.
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 experience developing 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.
Benefits
- Medical, dental, and vision coverage, potentially including 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.
- 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.
- Based in Austin, Texas.
About Graphcore
Graphcore designs and sells Intelligence Processing Units (IPUs), systems, and a full software stack for training and inference of AI models in data centers and research labs. Revenue comes from hardware systems and associated software tools and services, used for workloads across NLP, computer vision, and graph neural networks. Founded in 2016 and headquartered in Bristol, the company is part of SoftBank Group and focuses on on‑prem and cloud deployments.