jobapplier

Senior Machine Learning Engineer (Large Systems)

Graphcore

Skills

Required

  • Machine Learning
  • Python / C++
  • ML models
  • ML experiments
  • performance bottlenecks
  • software development
  • Data Science
  • Triton / CUDA
  • performance optimisation
  • distributed training

Preferred

  • frameworks
  • PyTorch / JAX
  • HPC systems
  • networking
  • Infiniband
  • NVLink
  • RoCE
  • cloud computing platforms
  • open-source
  • deep learning

Join Graphcore, a leading AI technology company, as a Senior Machine Learning Engineer. In this role, you will develop and optimize AI models for specialized hardware, work on large-scale systems, and collaborate with researchers in the rapidly-evolving field of AI. You will have the opportunity to innovate and differentiate Graphcore's technology while enjoying a flexible hybrid working model and a range of benefits including private medical insurance, pension, income protection, and opportunities for progression and development.

Building production systems with large language models

Keen to present, publish and deliver talks in the AI community

About Graphcore

  • Collaborer avec les équipes de recherche, de logiciels et de produits pour définir, construire et tester la prochaine génération de matériel d'IA de Graphcore.
  • If you're excited about advancing the next generation of AI models on cutting-edge hardware, we’d love to hear from you!

Responsibilities

  • Contribuer à l'avancement de la technologie de l'IA en développant et en optimisant de nouveaux modèles d'IA pour le matériel spécialisé.
  • Travailler sur des systèmes à grande échelle où la performance est critique pour le succès des projets, en identifiant les opportunités d'innovation.
  • Expertise in hardware-accelerated deep learning from model training to optimisation and evaluation
  • Enjoy cross-functional work collaborating with other teams
  • MLOps for Kubernetes-based clusters

Requirements

  • Capable of designing, executing and reporting from ML experiments
  • Well-developed understanding of performance bottlenecks and how to overcome them

Strong Python or C++ software development skills

  • Strong communicator - able to explain complex technical concepts to different audiences
  • Bachelor/Master's/PhD or equivalent experience in Machine Learning, Computer Science, Maths, Data Science, or related field
  • Ability to move quickly in a fast-moving field

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

Preferred

Proficiency in deep learning frameworks like PyTorch/JAX

  • 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

Every job at Graphcore