Graphcore - Senior Staff Engineer (ML Engineer)
Upload My Resume
Drop here or click to browse · PDF, DOCX, DOC, RTF, TXT
Requirements
• The ML QA team is composed of highly skilled software engineers with a strong focus on automation, software quality, and data-driven validation. The team works closely with industry-standard machine learning frameworks and models, contributing to upstream open-source projects and collaborating across the wider software organization. • Operating in a fast-paced environment, the team plays a critical role in ensuring reliability, performance, and maintainability across the ML software stack, helping to deliver robust and high-quality products to customers.
Responsibilities
• Benchmark ML models and frameworks, analysing results to identify regressions, performance bottlenecks, and correctness issues. • Work hands-on with industry-standard ML frameworks to validate functionality and performance across different execution environments. • Build and maintain automated testing and benchmarking pipelines targeting simulators, emulators, and physical hardware. • Collaborate closely with software teams to ensure adequate test coverage for new and existing features. • Develop tooling and scripts (primarily in Python) to support testing, benchmarking, and functional reporting. • Take ownership over aspects of our testing and infrastructure, owning the roadmap and driving innovation independently. • Candidate Profile • Candidate Profile • Essential: • 6+ years of experience working in Machine Learning or ML-adjacent engineering roles. • Strong foundation in core AI and ML concepts (e.g. neural networks, training vs inference, numerical precision, performance trade-offs). • Hands-on experience with one or more major ML frameworks such as PyTorch, TensorFlow, JAX, or similar. • Strong proficiency in Python for ML workflows, experimentation, and automation. • Experience designing, running, and analysing ML benchmarks or experiments. • Experience working in Linux environments. • Strong analytical and debugging skills, with the ability to reason about model behaviour and system performance. • Bachelor/Master's/PhD or equivalent experience in Computer Science, Maths, Machine Learning, Data Science, or related field. • Desirable • Experience with MLOps pipelines, model deployment, or production ML systems. • Familiarity with performance analysis, profiling tools, or numerical accuracy validation. • Exposure to distributed training or inference systems. • Experience with hardware-accelerated ML, compilers, or system-level performance considerations. • Familiarity with CI/CD systems used for ML workflows. • Experience contributing to open-source ML frameworks or tooling.
Benefits
• In addition to a competitive salary, Graphcore offers flexible working, a generous annual leave policy, private medical insurance and health cash plan, a dental plan, pension (matched up to 5%), life assurance and income protection. We have a generous parental leave policy and an employee assistance programme (which includes health, mental wellbeing, and bereavement support). We offer a range of healthy food and snacks at our central Bristol office and have our own barista bar! We welcome people of different backgrounds and experiences; we’re committed to building an inclusive work environment that makes Graphcore a great home for everyone. We offer an equal opportunity process and understand that there are visible and invisible differences in all of us. We can provide a flexible approach to interview and encourage you to chat to us if you require any reasonable adjustments. • Applicants for this position must hold the right to work in the UK. Unfortunately at this time, we are unable to provide visa sponsorship or support for visa applications
No credit card. Takes 10 seconds.