Senior ML Engineer: Scale AI Models on Custom Hardware

EngineersOfAI; Województwo pomorskie; On-site; PLN 260,400 - 474,400; Full time

Job summary

Graphcore is hiring Senior and Staff Machine Learning Engineers to advance AI models on our specialised hardware. You will develop and optimise models for large-scale systems and collaborate with Software and Research teams to push performance and efficiency.

You'll work on state-of-the-art AI workloads across platforms, with opportunities to influence the next generation of AI accelerators and contribute to real-world applications at scale.

Qualifications

• Bachelor/Master's/PhD or equivalent in ML, CS, Maths, DS, or related field.
• Proficiency in deep learning frameworks like PyTorch or JAX.
• Strong Python or C++ software development skills.
• Expertise in deep learning from model training to optimisation and evaluation.
• Ability to design, execute and report 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 topics.

Responsibilities

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

Skills

PyTorch/JAX

Python

C++

Deep learning

ML experimentation

Performance optimization

Education

Bachelor/Master/PhD in ML/CS/Math/Data Science

Job description

Graphcore is hiring Senior and Staff Machine Learning Engineers to advance AI models on our specialised hardware. You will develop and optimise models for large-scale systems and collaborate with Software and Research teams to push performance and efficiency.

You'll work on state-of-the-art AI workloads across platforms, with opportunities to influence the next generation of AI accelerators and contribute to real-world applications at scale.