Staff Machine Learning Engineer (Large Systems)

Graphcore
Cambridge, United Kingdom
3 weeks ago
Job Type
Permanent
Work Pattern
Full-time
Work Location
On-site
Seniority
Senior
Education
Degree
Posted
6 May 2026 (3 weeks ago)

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

As a Senior Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. You will work on large scale systems where performance is critical to the success of our projects. Working closely with the Software development and Research teams, you will play a critical role in identifying opportunities to innovate and differentiate Graphcore’s technology. We seek engineers with strong technical skills and an understanding of AI model implementation at scale, eager to make a tangible impact in this rapidly evolving field.


The Team

The Applied AI team’s role is to be proxies for our customers, we need to understand the latest AI models, applications, and software to ensure that Graphcore’s technology works seamlessly with the AI ecosystem and at scale. We build reference applications, contribute to key software libraries e.g. optimising kernels for efficiency on our hardware, and collaborate with the Research team to develop and publish novel ideas in domains such as efficient compute, model scaling and distributed training and inference of AI models for multiple modalities and applications.

If you're excited about advancing the next generation of AI models on cutting-edge hardware, we’d love to hear from you!

Responsibilities and Duties

  • Implement latest machine learning models and optimise them for performance and accuracy, scaling to 1000s of accelerators.
  • Test and evaluate new internal software releases, provide feedback to software engineering teams, make necessary code fixes, and conduct code reviews.
  • Benchmark models and key ML techniques to identify performance bottlenecks and improve model efficiency.
  • Design and conduct experiments on novel AI methods, implement them and evaluate results.
  • Collaborate with Research, Software, and Product teams to define, build, and test Graphcore’s next generation of AI hardware.
  • Engage with AI community and keep in touch with the latest developments in AI.

Candidate Profile

Essential:

  • Bachelor/Master's/PhD or equivalent experience in Machine Learning, Computer Science, Maths, Data Science, or related field.
  • Proficiency in deep learning frameworks like PyTorch/JAX.
  • Strong Python or C++ software development skills
  • Expertise in deep learning from model training to optimisation and evaluation.
  • Experience in distributed training or inference of ML models across 64+ accelerators.
  • Capable of designing, executing and reporting from ML experiments.
  • Developed deep understanding of performance bottlenecks and how to overcome them.
  • Ability to move quickly in a dynamic environment
  • Enjoy cross-functional work collaborating with other teams.
  • Strong communicator - able to explain complex technical concepts to different audiences.

Desirable:

  • Experience in one or more of:
    • MLOps for Kubernetes-based clusters
    • Building production systems with large language models
    • Efficient computing based on low-precision arithmetic.
  • Experience writing C++/Triton/CUDA kernels for performance optimisation of ML models.
  • Familiarity with HPC systems and networking including Infiniband, NVLink, RoCE technologies.
  • Have contributed to open-source projects or published research papers in relevant fields.
  • Knowledge of cloud computing platforms.
  • Keen to present, publish and deliver talks in the AI community.

Benefits

In addition to a competitive salary, Graphcore offers flexible working, a generous annual leave policy, private medical insurance and health cash plan, a dental plan, pension (matched up to 5%), life assurance and income protection. We have a generous parental leave policy and an employee assistance programme (which includes health, mental wellbeing, and bereavement support). We offer a range of healthy food and snacks at our central Bristol office and have our own barista bar! We welcome people of different backgrounds and experiences; we’re committed to building an inclusive work environment that makes Graphcore a great home for everyone. We offer an equal opportunity process and understand that there are visible and invisible differences in all of us. We can provide a flexible approach to interview and encourage you to chat to us if you require any reasonable adjustments.

Applicants for this position must hold the right to work in the UK. Unfortunately at this time, we are unable to provide visa sponsorship or support for visa applications

Related Jobs

View all jobs
Spotlight

Senior Processor Architect

Fractile London, United Kingdom
Spotlight

Semiconductor Test Engineering Team Leader

Fractile Bristol, United Kingdom
On-site

Staff Engineer (ML Engineer)

Graphcore Cambridge, United Kingdom
On-site

Staff Engineer (ML Engineer)

Graphcore London, United Kingdom
On-site

Staff Engineer (ML Engineer)

Graphcore Bristol, United Kingdom
On-site

Staff Software Engineer (ML QA)

Graphcore Cambridge, United Kingdom
On-site

Staff Software Engineer (ML QA)

Graphcore Bristol, United Kingdom
On-site

Staff Software Engineer (ML QA)

Graphcore London, United Kingdom
On-site

Industry Insights

Discover insightful articles, industry insights, expert tips, and curated resources.

Where to Advertise Semiconductor Jobs in the UK (2026 Guide)

Where to advertise semiconductor jobs UK in 2026: the specialist boards, academic channels and community routes that actually reach IC, fab and EDA talent. The candidate pool is one of the smallest and most specialised in any engineering discipline — spanning IC design engineers, process engineers, fab technicians, EDA tool developers, compound semiconductor physicists and power electronics specialists. General job boards are largely ineffective for semiconductor hiring. The community is tight-knit, highly academic in its roots and concentrated around a small number of university groups, fab facilities and design centres. Specialist boards, academic channels and direct community engagement are the primary sourcing strategies that work. This guide, published by SemiconductorJobs.co.uk, covers where to advertise semiconductor roles in the UK in 2026, how the main platforms compare, what employers should expect to pay, and what the data says about hiring across different role types.

Semiconductor Jobs UK 2026: What to Expect Over the Next 3 Years

Semiconductor Jobs UK 2026: roles, salaries and the hiring trends shaping UK semiconductor recruitment over the next three years — what to expect now. Semiconductors are the foundational technology of the modern world. Every smartphone, electric vehicle, data centre, medical device, satellite, and AI accelerator depends on them. And yet for much of the past decade, the strategic importance of semiconductor design and manufacturing was something that governments, investors, and employers took largely for granted — until supply chain crises, geopolitical tensions, and the insatiable compute demands of artificial intelligence made the vulnerability of global semiconductor supply chains impossible to ignore. The response has been significant and sustained. The UK's National Semiconductor Strategy, the US CHIPS Act, the EU Chips Act, and parallel investment programmes across Japan, South Korea, and Taiwan have collectively committed hundreds of billions of pounds to semiconductor research, design, and manufacturing capability. In the UK specifically, that investment is beginning to translate into real hiring — across compound semiconductor manufacturing, chip design, semiconductor equipment, advanced packaging, and the growing ecosystem of fabless design companies that are choosing Britain as their base. For job seekers, the semiconductor jobs market of 2026 represents an opportunity that is more commercially urgent, more geographically distributed, and more technically diverse than at any previous point in the sector's UK history. The roles being created span the full semiconductor value chain — from fundamental materials research and process engineering through chip design, verification, and the software that makes silicon useful. The candidates who will thrive over the next three years are those who understand where that value chain is being built, which technical areas are attracting the most investment, and how to position their skills at the intersection of the sector's greatest needs. This article breaks down what the UK semiconductor jobs market is likely to look like through to 2028 — covering the titles emerging right now, the technologies driving employer demand, the skills that will matter most, and how to position your career at the leading edge of one of the most strategically important technology sectors in the UK economy.