Staff Engineer ML Engineer

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

Job Summary

Applicants for this role should have strong experience working with machine learning systems and frameworks, along with a solid understanding of core AI concepts and model behaviour. The role centres on testing, validating, and benchmarking a complex ML software stack, with a particular focus on performance, reliability, and correctness across modern AI workloads.

The ideal candidate is an experienced ML engineer who understands how contemporary models are trained and executed, and who has hands-on experience debugging functional and performance issues in ML systems. This person will be comfortable working with industry-standard frameworks and state-of-the-art models, bringing them up on internal infrastructure, and collaborating closely with software and hardware teams in a technically demanding environment spanning ML frameworks, infrastructure, and AI accelerator hardware.

The Team

The ML QA team is composed of highly skilled software engineers with a strong focus on automation, software quality, and data-driven validation. The team works closely with industry-standard machine learning frameworks and models , contributing to upstream open-source projects and collaborating across the wider software organization.

Operating in a fast-paced environment, the team plays a critical role in ensuring reliability, performance, and maintainability across the ML software stack, helping to deliver robust and high-quality products to customers.

Responsibilities and Duties

• Benchmark ML models and frameworks, analysing results to identify regressions, performance bottlenecks, and correctness issues. • Work hands-on with industry-standard ML frameworks to validate functionality and performance across different execution environments. • Build and maintain automated testing and benchmarking pipelines targeting simulators, emulators, and physical hardware. • Collaborate closely with software teams to ensure adequate test coverage for new and existing features. • Develop tooling and scripts (primarily in Python) to support testing, benchmarking, and functional reporting. • Take ownership over aspect