AI Research Engineer

Contract type
Ongoing
Work mode
Remote · United Kingdom
Experience
3 years

Job description

Research engineer role at Graphcore contributing to AI research, investigating new ideas on important AI/ML problems and translating them into scalable implementations. The role combines strong machine learning experience with practical engineering skills, particularly lower-level programming and performance optimisation for hardware efficiency.

At a glance

Work mode
Remote · United Kingdom

What you'll do

  • Generate AI/ML ideas, design experiments, implement them & evaluate results.
  • Prepare, submit & present your work to AI conferences and workshops.
  • Provide technical insight to internal teams by designing experiments and delivering clear, actionable reports.
  • Collaborate with researchers, silicon and software engineers at Graphcore to help define, build and test Graphcore's next generation of AI hardware.

What we're looking for

  • Master's, PhD or equivalent experience in a technical discipline (e.g., Maths, Statistics, Computer Science, Physics, Chemistry).
  • Python programming in a modern deep learning framework, e.g. PyTorch or JAX.
  • Familiar with deep learning fundamentals: models, optimisation, evaluation and scaling.
  • Capable of designing, executing and reporting from ML experiments.
  • Lower-level programming for hardware efficiency, e.g. C++/CUDA/Triton.
  • Practical familiarity with hardware capabilities for deep learning – threads, caches, vector & matrix engines, data dependencies, bus widths and throttling.
  • Practical familiarity with software stacks for deep learning – compilation, kernel fusion, XLA/ATen ops, streams, and asynchronous execution.

Nice to have

  • Mathematics skills to support the above: calculus, probability theory and linear algebra.
  • Experience submitting papers to international scientific conferences or workshops.

Your team

Graphcore Research

Participates in both fundamental and applied research, to characterise the computational requirements of machine intelligence and to demonstrate how hardware can drive the next generation of innovative AI models. Publishes at leading AI/ML conferences (NeurIPS, ICML, ICLR) as well as specialist workshops, and collaborates with other research teams and organisations across the world. Based across London, Cambridge and Bristol.

Benefits

  • Competitive salary
  • Flexible working
  • Generous annual leave policy
  • Private medical insurance and health cash plan
  • Dental plan
  • Pension (matched up to 5%)
  • Life assurance and income protection
  • Generous parental leave policy
  • Employee assistance programme (which includes health, mental wellbeing, and bereavement support)
  • Range of healthy food and snacks at our central Bristol office and our own barista bar

Apply

About Graphcore

Building the future of AI compute

A team of semiconductor, software and AI experts, with deep experience in creating the complete AI compute stack - from silicon and software to infrastructure at datacenter scale. As part of the SoftBank Group, backed by significant long-term investment, delivering key technology into the fast-growing SoftBank AI ecosystem.

Key facts

  • AI compute / semiconductors
  • Part of the SoftBank Group
  • Backed by significant long-term investment
  • Delivering key technology into the fast-growing SoftBank AI ecosystem
  • Expanding teams around the world
  • Based across London, Cambridge and Bristol

What they value

  • Supportive and collaborative team
  • Inclusive work environment
  • Equal opportunity process