Impress employers and recruiters.
Choose from hundreds of resume examples.
Impress employers and recruiters.
Choose from hundreds of resume examples.
Target Resume Now
About the Role
As a Senior Machine Learning Engineer in the Applied AI team, you will advance AI technology by developing and optimizing AI models for specialized hardware. You will work on large-scale systems where performance is critical, collaborating with Software Development and Research teams to innovate and differentiate Graphcore’s technology. We seek engineers with strong technical skills and an understanding of AI model implementation at scale, eager to make a tangible impact in this rapidly evolving field.
Responsibilities
- Implement and optimize the latest machine learning models for performance and accuracy, scaling to thousands 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 stay updated on the latest developments in AI.
Requirements
- Bachelor/Master's/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 optimization and evaluation.
- Experience in distributed training or inference of ML models across 64+ accelerators.
- 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.
- Experience in MLOps for Kubernetes-based clusters.
- Experience building production systems with large language models.
- Experience with efficient computing based on low-precision arithmetic.
- Experience writing C++/Triton/CUDA kernels for performance optimization of ML models.
- Familiarity with HPC systems and networking including Infiniband, NVLink, RoCE technologies.
- Have contributed to open-source projects or published research papers in relevant fields.
- Knowledge of cloud computing platforms.
- Keen to present, publish and deliver talks in the AI community.
- Must hold the right to work in the UK.
- Unable to provide visa sponsorship or support for visa applications.
Skills
- Machine Learning
- Computer Science
- Maths
- Data Science
- PyTorch
- JAX
- Python
- C++
- Deep Learning
- MLOps
- Kubernetes
- Large Language Models
- Low-precision arithmetic
- Triton
- CUDA
- HPC systems
- Infiniband
- NVLink
- RoCE technologies
- Cloud computing
Location
- Bristol
Work Type
- Flexible working
Experience Level
- Senior
- 64+ accelerators
- 1000s of accelerators
Education Level
- Bachelor's Degree
- Master's Degree
- PhD
Benefits
- Competitive salary
- Flexible working
- Generous annual leave policy
- Private medical insurance
- Health cash plan
- Dental plan
- Pension (matched up to 5%)
- Life assurance
- Income protection
- Generous parental leave policy
- Employee assistance programme (health, mental wellbeing, and bereavement support)
- Healthy food and snacks
- Barista bar
About the Company
- Graphcore is building the future of AI compute.
- We are a team of semiconductor, software, and AI experts with deep experience in creating the complete AI compute stack.
- As part of the SoftBank Group, backed by significant long-term investment, we are delivering key technology into the fast-growing SoftBank AI ecosystem.
- Graphcore is expanding its teams around the world to meet the vast and exciting AI opportunity.
- We are bringing together the brightest minds to solve the toughest problems, in a place where everyone has the opportunity to make an impact on the company, our products, and the future of artificial intelligence.
Equal Opportunity
- We welcome people of different backgrounds and experiences; we’re committed to building an inclusive work environment that makes Graphcore a great home for everyone.
- We offer an equal opportunity process and understand that there are visible and invisible differences in all of us.
- We can provide a flexible approach to interview and encourage you to chat to us if you require any reasonable adjustments.