Senior ML Engineer — Optimizing AI Models at Scale

EngineersOfAI; Greater London; On-site; GBP 110,000 - 140,000; Full time

Job summary

Graphcore, based in London, is seeking a Senior Machine Learning Engineer for its Applied AI team to advance AI models on our specialised hardware. You will develop and optimise models, work with software and research teams, and push performance and scalability across large-scale systems.

You will contribute to scaling experiments, benchmarking, and deploying techniques that improve efficiency, accuracy, and throughput on Graphcore hardware, while staying aligned with the latest AI advances.

Qualifications

• Bachelor/Master’s/PhD or equivalent experience in ML/CS/Data Science.
• Proficiency in PyTorch or JAX.
• Strong Python or C++ software development skills.
• Expertise in deep learning from model training to optimisation and evaluation.
• Capable of designing, executing and reporting from ML experiments.
• Developed deep understanding of performance bottlenecks and how to overcome them.
• 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.

Responsibilities

• Implement latest machine learning models and optimise them for performance and accuracy, scaling to thousands 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 Graphcore’s next generation of AI hardware.
• Engage with AI community and keep in touch with the latest developments in AI.

Skills

Deep learning

Python

C++

Communication

Experiment design

Performance optimization

Cross-functional teamwork

Education

Bachelor's/Master's/PhD in ML/CS/Data Science

Tools

PyTorch

JAX

Job description

Graphcore, based in London, is seeking a Senior Machine Learning Engineer for its Applied AI team to advance AI models on our specialised hardware. You will develop and optimise models, work with software and research teams, and push performance and scalability across large-scale systems.

You will contribute to scaling experiments, benchmarking, and deploying techniques that improve efficiency, accuracy, and throughput on Graphcore hardware, while staying aligned with the latest AI advances.