Staff Engineer (ML Engineer)

EngineersOfAI; Bristol; On-site; GBP 60,000 - 85,000; Full time

Benefits offered by this job

Unlimited annual leave

Up to 5% matched pension

Phantom equity

True flexibility in work location

Free food and an on-site barista

Health cash plan

Income protection

Life assurance

Job summary

EngineersOfAI in Bristol is looking for a dedicated Machine Learning Validation Engineer to verify the ML stack that translates accelerator hardware into dependable AI performance. In this role, you will ensure ML models function reliably across various frameworks and execution environments while designing intricate benchmarks.

The ideal candidate possesses extensive ML engineering experience, strong Python skills, and hands-on knowledge of ML frameworks like PyTorch and TensorFlow. Benefits include unlimited annual leave, matched pension, equity opportunities, and more.

Qualifications

• Strong experience in Machine Learning or ML-adjacent software engineering roles.
• A solid grasp of neural networks, training, inference, numerical precision and performance trade-offs.
• Hands-on experience with PyTorch, TensorFlow, JAX, Triton, or similar ML frameworks and tools.
• Strong Python skills for automation, experimentation, benchmarking and reporting.
• Experience designing, running and analysing ML benchmarks or model experiments.
• Confident debugging skills in Linux, with curiosity about model behaviour and system performance.

Responsibilities

• Validate the ML stack that turns accelerator hardware into trusted AI performance.
• Test, benchmark and validate complex systems before they reach customers.
• Run open source models and build automated benchmarking pipelines.
• Create targeted tests for low-level ML behaviour.

Skills

Machine Learning experience

Neural networks understanding

Python skills

Debugging in Linux

Experience with ML frameworks

Tools

PyTorch

TensorFlow

JAX

Triton

Job description

About the job

Validate the ML stack that turns accelerator hardware into trusted AI performance.

This role sits where modern ML models meet Graphcore’s software and hardware stack. You will test, benchmark and validate complex systems before they reach customers.

Your work will expose regressions, correctness issues and performance limits across frameworks, models and execution environments. You will help teams understand what is working, what is breaking and why.

You will run open source models, build automated benchmarking pipelines and create targeted tests for low level ML behaviour. That includes numerical precision, quantisation, attention mechanisms, distributed execution and model subgraphs.

This is a role for someone who wants to stay close to how AI systems really work. You will not design new models, but you will make them run reliably on ambitious infrastructure.

The team and culture

The ML QA team is where Graphcore’s ML software stack comes together for validation. Work spans unit tests, full model benchmarks, distributed workloads, simulators, emulators and hardware targets.

Engineers are expected to take ownership, question assumptions and improve how testing is done. The team moves quickly, but decisions are grounded in evidence, benchmark data and technical discussion.

You will work closely with software, infrastructure and hardware teams. Squads organise around priorities, with space for engineers to shape roadmaps and raise the quality bar.

What we’re looking for

• Strong experience in Machine Learning or ML-adjacent software engineering roles.
• A solid grasp of neural networks, training, inference, numerical precision and performance trade-offs.
• Hands-on experience with PyTorch, TensorFlow, JAX, Triton or similar ML frameworks and tools.
• Strong Python skills for automation, experimentation, benchmarking and reporting.
• Experience designing, running and analysing ML benchmarks or model experiments.
• Confident debugging skills in Linux, with curiosity about model behaviour and system performance.

Benefits

• Unlimited annual leave
• Up to 5% matched pension
• Phantom equity - share in Graphcore's success
• True flexibility in how and where you work
• Office spaces designed for collaboration
• Free food and an on-site barista
• Health cash plan
• Income protection
• Life assurance
• Along with other benefits you can choose from (private medical insurance, dental plan etc)

We welcome people from all backgrounds and experiences and are committed to building an inclusive environment where everyone can do their best work.

We’re an equal opportunity employer and recognise that everyone brings different strengths and perspectives. If you need any adjustments during the interview process, just let us know - we’re happy to support you.