Staff Machine Learning Engineer (Large Systems)
As a Staff Machine Learning Engineer in the Applied AI team at Graphcore, you will play a pivotal role in advancing AI technology by developing and optimizing machine learning models tailored to our specialized hardware. This position focuses on large-scale systems where performance is critical to project success. Collaborating closely with Software Development and Research teams, you will identify opportunities to innovate and differentiate Graphcore's technology.
Your primary responsibilities will include implementing the latest machine learning models and optimizing them for performance and accuracy, scaling to thousands of accelerators. You will test and evaluate new internal software releases, provide feedback to software engineering teams, make necessary code fixes, and conduct code reviews. Additionally, you will benchmark models and key machine learning techniques to identify performance bottlenecks and improve model efficiency. Designing and conducting experiments on novel AI methods, implementing them, and evaluating results will also be key aspects of your role.
The ideal candidate will possess a Bachelor's, Master's, or PhD in Machine Learning, Computer Science, Mathematics, Data Science, or a related field. Proficiency in deep learning frameworks such as PyTorch or JAX, along with strong Python or C++ software development skills, is essential. Expertise in deep learning from model training to optimization and evaluation, experience in distributed training or inference of machine learning models across 64 or more accelerators, and a deep understanding of performance bottlenecks and strategies to overcome them are required. The ability to move quickly in a dynamic environment, enjoy cross-functional collaboration, and effectively communicate complex technical concepts to diverse audiences is also important.
Graphcore offers a competitive salary, flexible working arrangements, a generous annual leave policy, private medical insurance, a health cash plan, a dental plan, pension matched up to 5%, life assurance, and income protection. Additional benefits include 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 its central Bristol office, including an in-house barista bar.
Joining Graphcore means becoming part of a team at the forefront of the machine intelligence revolution, enabling innovators from all industries to build AI-native products that expand human potential. The company is committed to building an inclusive work environment where everyone has the opportunity to make an impact on the company, its products, and the future of artificial intelligence.