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