Senior ML Engineer: AI Hardware & Model Optimization

Graphcore Early Careers; Greater London; On-site; GBP 90,000 - 120,000; Full time

Benefits offered by this job

Flexible working

Annual leave policy

Private medical insurance

Health cash plan

Dental plan

Pension (matched up to 5%)

Life assurance

Income protection

Parental leave

Employee assistance programme

Job summary

Graphcore in the United Kingdom seeks a Senior Machine Learning Engineer for the Applied AI team to develop and optimise AI models for our specialised hardware.

You will implement state-of-the-art models, benchmark performance on large-scale accelerators, and collaborate with Research, Software, and Product teams to push the boundaries of AI compute.

The role emphasizes scalable ML, experimentation, and engaging with the AI community to keep Graphcore at the forefront of AI hardware.

Qualifications

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

Skills

Python development

C++ development

ML experimentation

Cross-functional collaboration

Technical communication

Ability to move quickly

Education

Bachelor/Master’s/PhD or equivalent in ML/CS/Data Science

Tools

PyTorch

JAX

CUDA

Job description

Graphcore in the United Kingdom seeks a Senior Machine Learning Engineer for the Applied AI team to develop and optimise AI models for our specialised hardware.

You will implement state-of-the-art models, benchmark performance on large-scale accelerators, and collaborate with Research, Software, and Product teams to push the boundaries of AI compute.

The role emphasizes scalable ML, experimentation, and engaging with the AI community to keep Graphcore at the forefront of AI hardware.