Senior Engineer (ML Engineer)
Graphcore is seeking a Senior Machine Learning Engineer to join their ML QA team in Cambridge, UK. This role focuses on testing, validating, and benchmarking a complex ML software stack, emphasizing performance, reliability, and correctness across modern AI workloads. The successful candidate will collaborate closely with software and hardware teams to ensure the robustness and high quality of Graphcore's AI products.
Key responsibilities include benchmarking ML models and frameworks to identify regressions and performance bottlenecks, working hands-on with industry-standard ML frameworks to validate functionality, and building automated testing and benchmarking pipelines targeting various execution environments. The role also involves developing tooling and scripts, primarily in Python, to support testing and functional reporting, as well as collaborating with software teams to ensure adequate test coverage for new and existing features.
The ideal candidate should have experience in machine learning or related engineering roles, a strong foundation in core AI and ML concepts, and hands-on experience with major ML frameworks such as PyTorch, TensorFlow, or JAX. Proficiency in Python for ML workflows, experience designing and analyzing ML benchmarks, and familiarity with Linux environments are essential. A Bachelor's, Master's, or PhD in Computer Science, Mathematics, Machine Learning, Data Science, or a related field is required.
Graphcore offers a competitive salary, flexible working arrangements, a generous annual leave policy, private medical insurance, a health cash plan, a dental plan, pension matching up to 5%, life assurance, and income protection. Additional benefits include a generous parental leave policy, an employee assistance program covering health and mental wellbeing, and a range of healthy food and snacks at their central Bristol office, which features its own barista bar.
Graphcore is committed to building an inclusive work environment that welcomes people of different backgrounds and experiences. They offer an equal opportunity process and understand the importance of accommodating visible and invisible differences. The company provides a flexible approach to interviews and encourages candidates to discuss any reasonable adjustments they may require.