machine learning engineer

Graphcore builds AI compute technology, creating a complete stack from silicon and software to infrastructure at datacenter scale. As part of the SoftBank Group, the company delivers technology into the SoftBank AI ecosystem and focuses on advancing AI compute through specialized hardware and software solutions.

• 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, make necessary code fixes, 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.

• Bachelor/Master's/PhD or equivalent experience in Machine Learning, Computer Science, Maths, Data Science, or related field;
• Proficiency in deep learning frameworks like PyTorch/JAX;
• Strong Python or C++ software development skills;
• Expertise in deep learning from model training to optimisation and evaluation;
• Experience in distributed training or inference of ML models across 64+ accelerators;
• Ability to design, execute and report from ML experiments;
• Deep understanding of performance bottlenecks and how to overcome them;
• Ability to move quickly in a dynamic environment;
• Experience in cross-functional collaboration;
• Strong communication skills;
• Nice to have: MLOps for Kubernetes-based clusters, building production systems with large language models, efficient computing based on low-precision arithmetic, experience writing C++/Triton/CUDA kernels, familiarity with HPC systems and networking (Infiniband, NVLink, RoCE), contributions to open-source projects or published research papers, knowledge of cloud computing platforms, interest in presenting and publishing in the AI community.

• Competitive salary;
• Flexible working;
• Generous annual leave policy;
• Private medical insurance and health cash plan;
• Dental plan;
• Pension (matched up to 5%);
• Life assurance and income protection;
• Generous parental leave policy;
• Employee assistance programme;
• Healthy food and snacks at the central Bristol office;
• Barista bar.