Senior Machine Learning Engineer

About the Company

At Graphcore, we’re building the future of AI compute. We’re a team of semiconductor, software and AI experts, with deep experience in creating the complete AI compute stack - from silicon and software to infrastructure at datacenter scale.

Responsibilities

• Implement latest machine learning models and optimise them for performance and accuracy, scaling to 1000s of accelerators
• Test and evaluate new internal software releases, provide feedback to software engineering teams, and conduct code reviews
• Benchmark models and key ML techniques to identify performance bottlenecks and improve model efficiency
• Design and conduct experiments on novel AI methods, implement them and evaluate results
• Collaborate with Research, Software, and Product teams to define, build, and test next generation AI hardware
• Engage with the AI community and keep in touch with the latest developments in AI

Requirements

• Bachelor/Master's/PhD in Machine Learning, Computer Science, Maths, Data Science, or related field
• Proficiency in deep learning frameworks like PyTorch or JAX
• Strong Python or C++ software development skills
• Expertise in deep learning from model training to optimisation and evaluation
• Ability to design, execute, and report on ML experiments
• Deep understanding of performance bottlenecks and how to overcome them
• Strong communication skills to explain complex technical concepts

Preferred Qualifications

• Experience with MLOps for Kubernetes-based clusters
• Experience building production systems with large language models
• Experience writing C++, Triton, or CUDA kernels for performance optimisation
• Experience in distributed training or inference across 64+ accelerators
• Familiarity with HPC systems and networking (Infiniband, NVLink, RoCE)
• Contributions to open-source projects or published research papers
• Knowledge of cloud computing platforms