Senior Engineer (ML Engineer)
GraphcoreAI & company
Cambridge, United KingdomSenior
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.
- •Key Responsibilities Test, benchmark and validate complex systems before they reach customers.
- •Run open source models, build automated benchmarking pipelines and create targeted tests for low level ML behaviour.
- •Expose regressions, correctness issues and performance limits across frameworks, models and execution environments.
- •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
PythonPyTorchTensorFlowJAX
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
PythonPyTorchTensorFlowJAX