Systems Performance Engineer, AI Infrastructure

CommonAI Holdings Ltd; Cambridge; Hybrid; GBP 70,000 - 110,000; Full time

Benefits offered by this job

Pension

Benefits package

Cambridge office

Job summary

CommonAI CIC in Cambridge is seeking a researcher to design and run reproducible experiments on AI infrastructure, building measurement tools and observing real hardware and software behavior in LLM serving systems.

You will work with workload, simulation and uncertainty specialists to identify bottlenecks, explain performance variation and calibrate simulator models. Hybrid work with on-site Cambridge time; UK-wide applicants welcome.

Qualifications

• Experience designing reproducible experiments for AI hardware/software stacks.
• Ability to build and use measurement tools to collect traces, profiles and counters.
• Strong profiling skills across accelerator, memory and HPC workloads.

Responsibilities

• Design and run reproducible experiments from components to full LLM serving systems.
• Develop reusable measurement tools and collect traces, profiles and counters to characterise latency, throughput, memory behaviour and energy.
• Investigate bottlenecks and performance variation across layers, extend investigations through the stack.
• Ensure reproducible workload details and calibration data for simulator models.
• Collaborate with simulation and uncertainty specialists to choose measurements that resolve unknowns.

Skills

Experiment design

Measurement tools

Profiling expertise

AI hardware basics

Tools

Nsight

PyTorch Profiler

System profilers

Job description

About CommonAI

CommonAI CIC is a non-profit organisation built around collaborative engineering and is backed by a mix of UK Government and private funding in order to design, build and deploy innovative AI systems.

Our Scaling Inference Lab (SIL) is developing simulation grounded in real measurements to help the UK invest in promising AI infrastructure. We connect proposed hardware designs to realistic workloads, helping investors and engineers understand their potential, their limitations and the next experiment worth doing.

The Opportunity

Design and run experiments that reveal how AI infrastructure behaves, what limits its performance and how reliably that behaviour can be predicted. You will build measurement tools and turn observations of real hardware and software into evidence for SIL’s simulation platform and the investment decisions it supports.

Your work will range from targeted experiments on individual components to measurements of complete LLM serving systems. Working with workload, simulation and uncertainty specialists, you will investigate bottlenecks, explain performance variation and identify the meaningful effects.

We welcome deep expertise in one or more parts of the hardware and software stack. You will work with colleagues who bring complementary strengths, with scope to broaden your expertise as the team grows.

What you will do

- Design and run reproducible experiments, from individual hardware or software components to complete LLM serving systems, covering relevant workloads, configurations and operating conditions.

- Develop reusable measurement tools and collect traces, profiles and counters to characterise latency, throughput, memory behaviour and, where reliable measurement is possible, power and energy.

- Investigate bottlenecks, performance variation and unexpected behaviour across the layers your expertise covers, working with colleagues to extend the investigation through the stack.

- Take responsibility for reproducible workload details, software versions, configuration and measurement methods so results can be reproduced, compared and used to calibrate simulator models.

- Work with simulation and uncertainty specialists to choose measurements that resolve important unknowns and undertake curation of calibration and validation data.

- Help define the controls needed for experiments and use the resulting cluster-control interfaces, coordinating access and changes with colleagues responsible for the hardware.

An example early project would be to work with workload colleagues to establish a reproducible reference experiment and measurement dataset, investigating how performance changes under a selected configuration or load change.

- Experience investigating the performance of real computing systems and designing experiments that support rigorous conclusions.

- Depth in one or more relevant areas: accelerator profiling, inference runtimes, memory systems, distributed/HPC workloads, networking, or power and energy measurement.

- The ability to build maintainable automation and analysis tools in a suitable software language, and to learn unfamiliar profiling tools.

- Knowledge and care of experimental controls, repeatability, confounding effects and the limits of a measurement.

- Some experience using AI coding agents, alongside sound debugging and review habits.

LLM serving experience and familiarity with tools such as Nsight, PyTorch Profiler or system/network profilers would be useful. These are examples, not a requirement to know every tool. Strong expertise in one part of the stack is welcome; we can shape the role and recruit complementary strengths around it.

Why join us

- Help build UK AI capability and improve how investment and compute serve the public good.

- Build a network spanning UK hardware SMEs, major technology companies, venture and public investors, and leading universities.

- Shape a growing team with startup pace and close academic links.

- Use frontier models and coding agents with a substantial tooling budget.

• A collaborative and supportive work environment
• The opportunity to have a high impact in a growing organisation
• Competitive salary package, including pension and benefits
• Professional development opportunities
• Networking opportunities with influential figures from across the tech sector and academia
• A vibrant office environment located a few minutes’ walk away from Cambridge train station

Working with us

Hybrid, with regular time in Cambridge agreed individually around hardware work and collaboration; applicants across the UK are welcome. We are open to permanent employment, part-time arrangements and consultancy. We would like you to start as soon as practical and can be flexible for the right person.

We particularly welcome experienced engineers but will also consider earlier-career candidates with strong evidence of ability. Relevant postgraduate study is useful, but qualifications are secondary to experience and the quality of your work.

CommonAI CIC is an equal opportunity employer and is committed to creating an inclusive and diverse workplace.