jobapplier

Senior Machine Learning Engineer (Large Systems)

Graphcore

Skills

Required

  • Machine Learning
  • Python / C++
  • ML models
  • Data Science
  • ML experiments
  • frameworks
  • PyTorch / JAX
  • software development
  • MLOps
  • Kubernetes-based clusters
  • Triton / CUDA
  • performance optimisation

Preferred

  • deep learning
  • model training
  • evaluation
  • AI model implementation
  • performance bottlenecks

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 field. You will have the opportunity to innovate and differentiate Graphcore's technology while enjoying a flexible hybrid working model and a range of benefits.

Building production systems with large language models

Have contributed to open-source projects or published research papers in relevant fields

Responsibilities

  • Contribuer au développement et à l'optimisation de nouveaux modèles d'IA pour le matériel spécialisé.
  • 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.
  • Participer à la conception, à l'exécution et à la communication des résultats des expériences en apprentissage automatique.
  • We seek engineers with strong technical foundations who are curious and eager to understand and advance AI model implementation, at scale
  • Well-developed understanding of performance bottlenecks and how to overcome them
  • Enjoy cross-functional work collaborating with other teams

Requirements

  • Bachelor/Master's/PhD or equivalent experience in Machine Learning, Computer Science, Maths, Data Science, or related field
  • Capable of designing, executing and reporting from ML experiments
  • Strong communicator - able to explain complex technical concepts to different audiences
  • Proficiency in deep learning frameworks like PyTorch/JAX
  • Ability to move quickly in a fast-moving field
  • Strong Python or C++ software development skills

MLOps for Kubernetes-based clusters

  • 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
  • Keen to present, publish and deliver talks in the AI community

Preferred

  • Expertise in hardware-accelerated deep learning from model training to optimisation and evaluation
  • Experience in one or more of:
  • Efficient computing based on low-precision arithmetic

Every job at Graphcore