3 months ago
Responsibilities
- Own performance analysis across the full ML stack, from model execution to hardware utilization.
- Identify bottlenecks and regression trends through profiling, benchmarking, and deep system analysis.
- Drive cross-team optimizations involving ML frameworks, compilers, runtimes, and infrastructure.
- Design and maintain benchmarking environments for focused synthetic benchmarks and large-scale ML workloads.
- Produce performance reports that translate low-level findings into actionable insights.
- Ensure local optimizations improve global system performance rather than isolated components.
- Partner with engineering teams to guide performance improvements and validate their impact.
Requirements
- Strong programming skills in Python, C, or C++, particularly for performance-sensitive applications.
- Solid understanding of computer architecture, performance profiling, and low-level CPU, memory, and I/O behavior.
- Experience benchmarking and analyzing complex distributed systems.
- Familiarity with Linux-based development environments and tools.
- Strong problem-solving skills and ability to interpret and communicate performance data clearly.
- Passion for the work and ability to operate in uncertain and complex environments.
- Knowledge of ML frameworks, ideally PyTorch, and their performance characteristics is desirable.
- Experience analyzing performance in GPU-accelerated environments such as CUDA or ROCm is desirable.
- Familiarity with ML hardware performance and high-speed networking such as RoCE and RDMA is desirable.
- Familiarity with distributed computing frameworks and collective operations is desirable.
- Experience building performance-monitoring dashboards or visualizations using Grafana, Prometheus, or custom tooling is desirable.
- Exposure to performance regression tracking and CI pipelines for performance validation is desirable.
Benefits
- Annual leave policy
- Medical and dental health plans
- Gym card
- Employee pension matched up to 4%
- Flexible interview approach and reasonable adjustments available
- Inclusive and equal-opportunity work environment
Categories
About Graphcore
At Graphcore, we’re building the future of AI compute. We’re a team of semiconductor, software and AI experts, with deep experience in creating the complete AI compute stack - from silicon and software to infrastructure at datacenter scale. As part of the SoftBank Group, backed by significant long-term investment, we are delivering key technology into the fast-growing SoftBank AI ecosystem. To meet the vast and exciting AI opportunity, Graphcore is expanding its teams around the world. We are bringing together the brightest minds to solve the toughest problems, in a place where everyone has the opportunity to make an impact on the company, our products and the future of artificial intelligence.