Senior ML Engineer: AI Hardware & Scalable Models

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

Job summary

Graphcore in Cambridge is seeking a Senior Machine Learning Engineer for the Applied AI team to advance AI technology by developing and optimising models for our specialised hardware. You’ll work on large-scale systems where performance is critical, collaborating with software and research teams to push the state of the art.

You’ll implement models end-to-end, evaluate experiments, and contribute to scalable, efficient AI solutions across multiple modalities, while remaining engaged with the AI

Qualifications

• Bachelor/Master/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 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 1000s 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 the AI community and keep in touch with the latest developments in AI.

Skills

Machine Learning

Python

C++

Deep Learning

Model Training

Experimentation

Communication

Cross-functional teamwork

Performance optimization

Education

Bachelor/Master/PhD in ML/CS

Tools

PyTorch

JAX

CUDA

Triton

Job description

Graphcore in Cambridge is seeking a Senior Machine Learning Engineer for the Applied AI team to advance AI technology by developing and optimising models for our specialised hardware. You’ll work on large-scale systems where performance is critical, collaborating with software and research teams to push the state of the art.

You’ll implement models end-to-end, evaluate experiments, and contribute to scalable, efficient AI solutions across multiple modalities, while remaining engaged with the AI