Machine Learning Engineer

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you will play a critical role in finding opportunities to innovate and differentiate Graphcore's technology. We seek engineers with strong technical skills and an understanding of AI model implementation, eager to make a tangible impact in this rapidly evolving field.

The Team

The Applied AI team's role is to be proxies for our customers; we need to understand the latest AI models, applications, and software to ensure that Graphcore's technology works seamlessly with the AI ecosystem. We build reference applications, contribute to key software libraries (e.g., optimizing kernels for efficiency on our hardware), and collaborate with the Research team to develop and publish novel ideas in domains such as efficient compute, model scaling, and distributed training and inference of AI models for multiple modalities and applications.

If you're excited about advancing the next generation of AI models on cutting-edge hardware, we'd love to hear from you!

Responsibilities and Duties

1. Implement the latest machine learning models and optimise them for performance and accuracy, scaling to thousands of accelerators.
2. Test and evaluate new internal software releases, provide feedback to software engineering teams, make vital code fixes, and conduct code reviews.
3. Benchmark models and key ML techniques to identify performance bottlenecks and improve model efficiency.
4. Design and conduct experiments on novel AI methods, implement them, and evaluate results.
5. Collaborate with Research, Software, and Product teams to define, build, and test Graphcore's next generation of AI hardware.
6. Engage with the AI community and stay updated with the latest developments in AI.

Candidate Profile

Essential skills:

1. Bachelor's, Master's, PhD, or equivalent experience in Machine Learning, Computer Science, Maths, Data Science, or related fields.
2. Proficiency in deep learning frameworks like PyTorch or JAX.
3. Strong Python software development skills (knowledge of C++ or other languages is a plus).
4. Familiarity with deep learning fundamentals such as models, optimisation, evaluation, and scaling.
5. Experience in designing, executing, and reporting ML experiments.
6. Ability to work quickly and effectively in a dynamic environment.
7. Enjoy cross-functional collaboration with other teams.
8. Strong communication skills, capable of explaining complex technical concepts to diverse audiences.
9. Experience in one or more areas such as distributed training of large-scale ML models, building production systems with large language models, efficient computing with low-precision arithmetic, or large generative models for language, vision, and other modalities.
10. Experience writing C++, Triton, or CUDA kernels for performance optimisation of ML models.
11. Contributions to open-source projects or published research papers in relevant fields.
12. Knowledge of cloud computing platforms.
13. Keen to present, publish, and deliver talks in the AI community.

In addition to a competitive salary, Graphcore offers flexible working, a generous annual leave policy, private medical insurance, a health cash plan, dental plan, pension (matched up to 5%), life assurance, and income protection. We also provide a generous parental leave policy, an employee assistance programme (covering health, mental wellbeing, and bereavement support), healthy food and snacks at our Bristol office, and an on-site barista. We value diversity and are committed to creating an inclusive work environment. We offer flexible interview arrangements and reasonable adjustments upon request.

Applicants must have the right to work in the UK. We are currently unable to offer visa sponsorship or support.

Apply for this job

indicates a required field

First Name

Last Name

Email

Phone

Resume/CV

Enter manually

Accepted file types: pdf, doc, docx, txt, rtf

Have you added your full legal name and surname?

Do you have the legal right to work in the UK?

Please select your right to work status

If your right to work status is not included in the list, please share some context

LinkedIn Profile

How did you hear about this role?

UK Demographic Data

We take pride in our commitment to creating an inclusive and diverse workplace. As part of our recruitment process, we ask for confidential diversity data from all applicants. This data will be anonymised for statistical purposes only and will not impact your application. Your responses will remain confidential and will not be used in any way regarding your application. We are only using this data to improve our hiring process to be inclusive of all backgrounds.

What is your ethnicity?

Do you consider yourself to have a disability?