Senior ML Engineer - AI Hardware & Scale

EngineersOfAI; Bristol; On-site; GBP 90,000 - 140,000; Full time

Job summary

Graphcore, based in Bristol, is seeking a Senior Machine Learning Engineer for the Applied AI team. You will develop and optimize AI models for our specialized hardware, scaling to thousands of accelerators, and collaborate with Software and Research teams to push AI compute forward.

You should have strong ML fundamentals, Python or C++, and experience with PyTorch or JAX. You’ll design and run ML experiments, identify performance bottlenecks, and help deliver efficient kernels and distributed

Qualifications

• Bachelor/Master’s/PhD or equivalent in ML, CS, 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

Python

C++

PyTorch/JAX

ML Optimization

Experiment Design

Performance Bottlenecks

Cross-functional

Communication

Education

Bachelor/Master/PhD

Job description

Graphcore, based in Bristol, is seeking a Senior Machine Learning Engineer for the Applied AI team. You will develop and optimize AI models for our specialized hardware, scaling to thousands of accelerators, and collaborate with Software and Research teams to push AI compute forward.

You should have strong ML fundamentals, Python or C++, and experience with PyTorch or JAX. You’ll design and run ML experiments, identify performance bottlenecks, and help deliver efficient kernels and distributed