# Senior Machine Learning Engineer (Large Systems)

**Company**: Graphcore
**Location**: Gdańsk, Pomeranian Voivodeship
**Experience**: senior
**Job type**: full-time
**Salary**: PLN 260,400 - 352,200
**Category**: Engineering
**Industry**: Technology

**Apply**: https://job-boards.greenhouse.io/graphcore/jobs/8564387002?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_e42c91d5-be0

## Description

As a Senior Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware.

You will work on large scale systems where performance is critical to the success of our projects. Working closely with the Software development and Research teams, you will play a critical role in identifying opportunities to innovate and differentiate Graphcore’s technology.

The Applied AI team’s role is to be proxies for our customers, we need to understand the latest AI models, applications, and software to ensure that Graphcore’s technology works seamlessly with the AI ecosystem and at scale.

Responsibilities and Duties:

- Implement latest machine learning models and optimise them for performance and accuracy, scaling to 1000s 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 AI community and keep in touch with the latest developments in AI.

Candidate Profile: 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 deep learning from model training to optimisation and evaluation.

- 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.

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.

- 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.

Benefits: In addition to a competitive salary, Graphcore offers annual leave policy, medical and dental health plans, a gym card, and employee pension (matched up to 4%).

## Skills

### Required
- Machine Learning
- Computer Science
- Maths
- Data Science
- PyTorch
- JAX
- Python
- C++
- Deep Learning

### Nice to have
- MLOps
- Kubernetes
- Large Language Models
- Efficient Computing
- C++/Triton/CUDA kernels
- Distributed Training
- HPC systems
- Cloud Computing

---

Source: [Apply at job-boards.greenhouse.io](https://job-boards.greenhouse.io/graphcore/jobs/8564387002?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)