2026 Software Engineering Intern - ML Kernels & Runtime Team
Upload My Resume
Drop here or click to browse · PDF, DOCX, DOC, RTF, TXT
Requirements
• Courseworks or past experience in using ML frameworks, parallel programming, or code optimisation. • Exposure to math libraries such as MKL or OpenBLAS. • Knowledge of performance analysis tools.
Responsibilities
• Supporting the design and implementation of kernels for linear algebra and tensor ops (GEMM, batched GEMM, convolutions, reductions, elementwise and fused operations) in C++ • Profile and optimise for the next generation of AI hardware - threading, cache locality, memory layout, and kernel launch efficiency. • Support performance and correctness - add microbenchmarks, regression tests, numerics validation • Debug issues, resolve bugs and generally improve the quality and functionality of the product • You are open-minded and collaborative with interests in performance optimisation and memory-efficient designs, and you are looking to join a team of experts. • You are comfortable to discuss technical tradeoffs, receive feedback and iterate on solutions and you are drawn to technically challenging problems and use analyticals reasoning to navigate unfamiliar domains. • Essential: • Bachelor or Master's Degree in Computer Science, Maths, Machine Learning, Data Science, or related field • Experirence in C/C++11. • Familiarity with Python or scripting tools for automation and testing. • Understanding of linear algebra, numerical methods, or scientific computing. • Good problem-solving skills and ability to work collaboratively in a fast-paced environment.
Benefits
• In addition to a competitive salary, Graphcore offers flexible working, a generous annual leave policy, private medical insurance and health cash plan, a dental plan, pension (matched up to 5%), life assurance and income protection. We have a generous parental leave policy and an employee assistance programme (which includes health, mental wellbeing, and bereavement support). We offer a range of healthy food and snacks at our central Bristol office and have our own barista bar! 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.
Similar Jobs
No credit card. Takes 10 seconds.