Senior Engineer (ML Engineer)

Senior Engineer (ML Engineer) at Graphcore.

About the role You will ensure the reliability and performance of the Graphcore software stack by validating how it interacts with our AI accelerator hardware. This position focuses on testing and benchmarking complex systems to identify regressions and performance bottlenecks before deployment.

Key facts

• Location: Cambridge, UK
• Engagement: Full-time
• Team: ML QA

What you'll do

• Execute open source models and develop automated pipelines for benchmarking.
• Create specific tests for low-level ML operations including quantization, numerical precision, and distributed execution.
• Diagnose correctness issues and performance limits across various frameworks and model subgraphs.
• Collaborate with hardware, software, and infrastructure teams to improve testing methodologies.

Requirements

• Proven background in Machine Learning or related software engineering.
• Proficiency in Python for benchmarking, automation, and data analysis.
• Understanding of neural network training, inference, and performance trade-offs.
• Experience with ML frameworks such as JAX, PyTorch, TensorFlow, or Triton.
• Ability to debug complex issues within Linux environments.
• Experience designing and running ML model experiments.

Skills & tools

• Python
• PyTorch
• TensorFlow
• JAX
• Triton
• Linux

Benefits

• Benefits include unlimited annual leave, up to 5% matched pension, and phantom equity.
• The package also covers a health cash plan, income protection, life assurance, and optional private medical or dental insurance.
• The office provides free food and an on-site barista.
• Graphcore is part of the SoftBank Group and supports flexible working arrangements.

#J-18808-Ljbffr