Staff Engineer (ML Engineer)

📋 Description

• Validate the ML stack that turns accelerator hardware into trusted AI performance.
• Test, benchmark and validate complex systems before they reach customers.
• Expose regressions, correctness issues and performance limits across frameworks, models and
• Run open source models, build automated benchmarking pipelines and create targeted tests for low
• Work closely with software, infrastructure and hardware teams; contribute to roadmaps and raise

🎯 Requirements

• Strong experience in Machine Learning or ML-adjacent software engineering roles.
• 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