Senior Software Engineer
Graphcore is seeking a Senior Software Engineer to join their Machine Learning Quality Assurance (ML QA) team in Cambridge, UK. This role focuses on testing, validating, and benchmarking a complex machine learning 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 delivery of robust and high-quality AI products.
Key responsibilities include benchmarking machine learning models and frameworks to identify regressions, performance bottlenecks, and correctness issues. The engineer will work hands-on with industry-standard ML frameworks to validate functionality and performance across various execution environments. Additionally, the role involves building and maintaining automated testing and benchmarking pipelines targeting simulators, emulators, and physical hardware. Collaboration with software teams to ensure adequate test coverage for new and existing features is also essential.
The ideal candidate should have experience in machine learning or related engineering roles, with a strong foundation in core AI and ML concepts such as neural networks, training versus inference, numerical precision, and performance trade-offs. Proficiency in Python for ML workflows, experimentation, and automation is required, along with hands-on experience with major ML frameworks like PyTorch, TensorFlow, or JAX. Experience in designing, running, and analyzing ML benchmarks or experiments, as well as working in Linux environments, is also necessary. A Bachelor's, Master's, or PhD in Computer Science, Mathematics, Machine Learning, Data Science, or a related field is expected.
In addition to a competitive salary, Graphcore offers flexible working arrangements, a generous annual leave policy, private medical insurance, a health cash plan, a dental plan, and a pension scheme matched up to 5%. Additional benefits include life assurance, income protection, a generous parental leave policy, and an employee assistance program covering health, mental wellbeing, and bereavement support. The company provides a range of healthy food and snacks at their central Bristol office, including a barista bar.
Graphcore is committed to building an inclusive work environment that values diversity and welcomes people from different backgrounds and experiences. The company offers equal opportunity processes and understands the importance of accommodating visible and invisible differences among employees. Flexible approaches to interviews and reasonable adjustments are available to support all candidates.