Machine Learning Engineer

at  graphcore

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.

ESSENTIAL SKILLS:

• 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 software development skills (nice to have C++/other languages).
• Familiar with deep learning fundamentals: models, optimisation, evaluation and scaling.
• Capable of designing, executing and reporting from ML experiments.
• Ability to move quickly in a dynamic environment.
• Enjoy cross-functional work collaborating with other teams.
• Strong communicator - able to explain complex technical concepts to different audiences.

• Implement the 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 vital 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 Graphcore’s next generation of AI hardware.
• Engage with AI community and keep in touch with the latest developments in AI.