Senior Engineer (ML Engineer)
On-siteCambridge, 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 to help teams understand what is working, breaking, and why. Run open source models, build automated benchmarking pipelines, and create targeted tests for low-level ML behavior including numerical precision, quantization, attention mechanisms, distributed execution, and model subgraphs. Make models run reliably on ambitious infrastructure without designing new models. Work closely with software, infrastructure, and hardware teams to shape roadmaps and raise the quality bar using evidence and 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.