AI Research Engineer
Salary not listed • Cambridge, UK • Full-time
Posted 3mo ago
This job has been open for more than 60 days. It may no longer be actively hiring, so check with the company before putting a lot into an application.
Job Location
Cambridge, UK
Also open in Bristol, UK, London, UK
Tech Stack
Remote Work Policy
On-site
Employment Type
Full-time
Categories
AI Research Engineer
About the job
Graphcore is building the future of AI compute, from silicon and software to datacenter-scale infrastructure. As a Research Engineer, you will contribute to advancing AI research by investigating new ideas and pushing the limits on important AI/ML problems. This role combines strong machine learning experience with practical engineering skills to deliver impactful AI research, focusing on hardware-aware AI algorithms and AI-aware hardware developments. You will work closely with researchers to generate ideas, translate them into scalable implementations, and contribute to publications and projects that steer the future of AI hardware.
Responsibilities
- Generate AI/ML ideas, design experiments, implement them, and evaluate results.
- Prepare, submit, and present 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 to help define, build, and test next-generation AI hardware.
Requirements
- 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).
- Familiarity with deep learning fundamentals: models, optimisation, evaluation, and scaling.
- Capability to design, execute, and report 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, throttling).
- Practical familiarity with software stacks for deep learning (compilation, kernel fusion, XLA/ATen ops, streams, asynchronous execution).
- Mathematics skills: calculus, probability theory, and linear algebra.
- Experience submitting papers to international scientific conferences or workshops.
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
- Income protection
- Generous parental leave policy
- Employee assistance programme (health, mental wellbeing, bereavement support)
- Healthy food and snacks at the Bristol office
- Barista bar