Senior: PLN 260,400 - 352,200
Staff: PLN 350,700 - 474,400
About the job
Help scale state-of-the art AI models across thousands of accelerators.
As a Senior Machine Learning Engineer in the Applied AI team, you will contribute to advancing AI technology by developing and optimising new and existing AI models for our specialised hardware. You will work on large scale systems where performance is critical to the success of our projects.
Having visibility of the entire pipeline from novel accelerator hardware to state-of-the-art AI applications, and the software stack in-between, you will play a critical role in identifying opportunities to innovate and differentiate Graphcore’s technology.
Furthermore, you’ll work closely with researchers in a rapidly-evolving field, where even the most senior engineers are constantly learning and adapting to exciting new challenges.
We seek engineers with strong technical foundations who are curious and eager to understand and advance AI model implementation, at scale.
We currently have multiple opportunities available in the team at Senior and Staff level. We offer flexibility through a hybrid working model, from our modern office in Gdánsk.
If you're excited about advancing the next generation of AI models on cutting-edge hardware, we’d love to hear from you!
The team and culture
The Applied AI team’s role is to be proxies for our customers, we need to understand the latest AI models, applications, and software, as well as our own hardware and software stack, to ensure that Graphcore’s technology works seamlessly with the AI ecosystem and at scale.
Our work spans from low-level kernel development and optimisation for novel hardware through to implementing and scaling research-level algorithms for the latest AI models.
We collaborate with the Research team to develop and publish novel ideas in domains such as efficient compute, model scaling and distributed training and inference of AI models for multiple modalities and applications.
Responsibilities
- Implement and train state-of-the-art machine learning models and optimise for performance, accuracy and scalability across systems comprising 1000s of accelerators.
- Benchmark and profile ML models to identify performance bottlenecks.
- Develop deep understanding across the software stack in order to optimise kernel implementations.
- Test and evaluate new internal software releases, provide feedback to software engineering teams, make necessary code fixes, and conduct code reviews.
- Design and conduct experiments on novel AI methods 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.
What we’re looking for
Essential:
- 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 hardware-accelerated deep learning from model training to optimisation and evaluation.
- Capable of designing, executing and reporting from ML experiments.
- Well-developed understanding of performance bottlenecks and how to overcome them.
- Ability to move quickly in a fast-moving field.
- Enjoy cross-functional work collaborating with other teams.
- Strong communicator - able to explain complex technical concepts to different audiences.
Desirable:
- Experience in one or more of:
- MLOps for Kubernetes-based clusters
- Building production systems with large language models
- Efficient computing based on low-precision arithmetic.
- Experience writing C++/Triton/CUDA kernels for performance optimisation of ML models.
- Experience in distributed training or inference of ML models across 64+ accelerators.
- Familiarity with HPC systems and networking including Infiniband, NVLink, RoCE technologies, and cloud computing platforms.
- Have contributed to open-source projects or published research papers in relevant fields.
- Keen to present, publish and deliver talks in the AI community.
Benefits
- Flexible working: Balance your work and personal life with greater flexibility
- Comprehensive healthcare and dental cover: Premium medical and dental insurance, with the option to extend cover to your family
- Phantom equity: Share in Graphcore’s success
- Fitness and wellbeing support: A fully funded Medicover gym card, giving you access to a wide range of gyms, fitness classes and leisure centres
- Financial protection: Death and disability insurance, providing financial security for you and your loved ones
- Retirement savings: Build your financial future, with a 4% employer contribution from Graphcore, alongside your own contributions
- Employee Assistance Programme (EAP): Confidential advice and support for you and your family across a range of personal, financial and wellbeing matters
We welcome people from all backgrounds and experiences and are committed to building an inclusive environment where everyone can do their best work.
We’re an equal opportunity employer and recognise that everyone brings different strengths and perspectives. If you need any adjustments during the interview process, just let us know - we’re happy to support you.
Join the Team at Graphcore
Graphcore is one of the world’s leading innovators in Artificial Intelligence compute. It is developing hardware, software and systems infrastructure that will unlock the next generation of AI breakthroughs and power the widespread adoption of AI solutions across every industry.
As part of the SoftBank Group, Graphcore is a member of an elite family of companies responsible for some of the world’s most transformative technologies. Together, they share a bold vision: to enable Artificial Intelligence and ensure its benefits are accessible to everyone.
Graphcore’s teams are drawn from diverse backgrounds and bring a broad range of skills and perspectives. A melting pot of AI research specialists, silicon designers, software engineers and systems architects, Graphcore brings together deep expertise to solve complex problems and deliver meaningful progress in AI compute.
Sponsorship
Applicants must have the legal right to work in Poland. Unfortunately, we are unable to provide visa sponsorship for this role.
Skills Required
- Bachelor's, Master's or PhD in Machine Learning, Computer Science, Mathematics, Data Science or equivalent experience
- Proficiency with deep learning frameworks such as PyTorch or JAX
- Strong software development skills in Python or C++
- Expertise across model training, optimisation and evaluation for deep learning
- Experience in distributed training or inference across 64+ accelerators
- Ability to design, execute and report ML experiments
- Deep understanding of performance bottlenecks and mitigation strategies
- Strong communication and cross-functional collaboration skills
- Experience with MLOps for Kubernetes-based clusters
- Experience building production systems with large language models
- Experience writing C++/Triton/CUDA kernels for performance optimisation
- Familiarity with HPC systems and networking (Infiniband, NVLink, RoCE)
- Contributed to open-source projects or published relevant research
- Knowledge of cloud computing platforms
What the Team is Saying
Graphcore Compensation & Benefits Highlights
-
Healthcare Strength — Healthcare coverage includes private medical insurance, dental cover, a health cash plan, mental-wellbeing support, and in the U.S. medical, dental and vision options with mental-health services. Family coverage is available in some locations, indicating broad support even if dependents may involve additional cost.
-
Retirement Support — Retirement support features a UK pension match up to 5% alongside life assurance and income protection, with U.S. materials describing a 401(k) and disability and life insurance. These provisions signal solid employer-backed long-term security across major hubs.
-
Leave & Time Off Breadth — Time off and flexibility are positioned strongly, with generous parental leave and an unlimited annual-leave policy on the careers page plus flexible PTO noted in U.S. hiring materials. Flexible working hours and hybrid arrangements are also highlighted.
Graphcore Insights
What We Do
At Graphcore, we’re building the future of AI compute. We’re 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, we are delivering key technology into the fast-growing SoftBank AI ecosystem. To meet the vast and exciting AI opportunity, Graphcore is expanding its teams around the world. 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.
Why Work With Us
Our team is at the forefront of the machine intelligence revolution, enabling innovators from all industries to build AI-native products to expand human potential. What we do at Graphcore really makes a difference.
Gallery
Graphcore Offices
Hybrid Workspace
Employees engage in a combination of remote and on-site work.
At Graphcore, we value wellbeing and flexibility to support a healthy work/life balance. Our hybrid approach encourages office-based colleagues to work onsite three days a week, with trusted flexibility built on trust and transparency for everyone.
