Staff Engineer (ML Engineer)
On-siteLondon, England, United Kingdom
Job Summary
Validate the ML stack that turns accelerator hardware into trusted AI performance by testing, benchmarking, and validating complex systems before they reach customers. Expose regressions, correctness issues, and performance limits across frameworks, models, and execution environments while running open source models and building automated benchmarking pipelines. Create targeted tests for low-level ML behavior including numerical precision, quantization, attention mechanisms, distributed execution, and model subgraphs. Work closely with software, infrastructure, and hardware teams to ensure models run reliably on ambitious infrastructure without designing new models. This role sits where modern ML models meet Graphcore's software and hardware stack, focusing on evidence-based decisions grounded in benchmark data.
Required 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
Hiring someone like this?
Get your role in front of qualified candidates on Sorce.