Staff Engineer (ML Engineer)
GraphcoreAI & company
Bristol, United KingdomLead
Baillie Gifford
SoftBank Group
Ontario Teachers' Pension Plan
Fidelity International
Schroders
Draper Esprit
Data & AI
About the role
TL;DR
Validate ML stack for AI performance.
- •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.
- •Key Responsibilities Expose regressions, correctness issues and performance limits across frameworks, models and execution environments.
- •Help teams understand what is working, what is breaking and why.
- •Run open source models, build automated benchmarking pipelines and create targeted tests for low level ML behaviour.
- •Requirements 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.
Required skills
PythonTensorFlowPyTorchJAXLinux
Domain expertise
ai
Benefits & perks
Unlimited annual leave, Up to 5% matched pension, Phantom equity, 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
Tech stack
PythonTensorFlowPyTorchJAXLinux