Senior Machine Learning Engineer (Large Systems)
As a Senior Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimizing AI models tailored to our specialized hardware. You will work on large-scale systems where performance is critical to the success of our projects. Working closely with the Software Development and Research teams, you will play a critical role in identifying 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.
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 like PyTorch or JAX, along with strong Python or C++ software development skills, is essential. Experience in distributed training or inference of machine learning models across 64 or more accelerators, as well as a deep understanding of performance bottlenecks and how to overcome them, is highly desirable.
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 matched up to 5%. We also provide life assurance, income protection, a generous parental leave policy, and an employee assistance program that includes health, mental wellbeing, and bereavement support. Our central Bristol office offers a range of healthy food and snacks, along with our own barista bar.
At Graphcore, we are committed to building an inclusive work environment that makes our company a great home for everyone. We welcome people of different backgrounds and experiences and offer an equal opportunity process. We understand that there are visible and invisible differences in all of us and can provide a flexible approach to interviews, encouraging you to chat with us if you require any reasonable adjustments.