Senior Machine Learning Engineer

Graphcore is a SoftBank-owned AI-compute company developing AI processors, systems, and software for machine-learning workloads.

Bristol, United Kingdom
About Graphcore

Graphcore develops Intelligence Processing Units (IPUs) and the Poplar SDK for building and running machine-learning applications. It continues operating under the Graphcore name as a wholly owned SoftBank Group subsidiary.

View jobs by Graphcore

Skills

Candidate Availability

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You will implement and optimise machine-learning models for performance and accuracy at large scale. You will test software releases, benchmark models, identify bottlenecks, design experiments, review code, and collaborate to define, build, and test AI hardware.

Requirements

  • Proficiency in PyTorch or JAX
  • Strong Python or C++ software development skills
  • Expertise in deep learning model training, optimisation and evaluation
  • Ability to design, execute and report machine learning experiments
  • Understanding of performance bottlenecks and their solutions
  • Cross-functional collaboration
  • Technical communication

Responsibilities

  • Implement and optimise machine learning models for performance and accuracy
  • Test and evaluate internal software releases
  • Provide feedback and code fixes
  • Conduct code reviews
  • Benchmark models and machine learning techniques
  • Identify performance bottlenecks and improve model efficiency
  • Design, conduct and evaluate AI experiments
  • Define, build and test next-generation AI hardware
  • Engage with the AI community

Benefits

  • Annual leave
  • Medical health plan
  • Dental health plan
  • Gym card
  • Up to 4% matched pension