Systems Performance Engineer, AI Infrastructure

CommonAI C.I.C.; Town of Cambridge; Hybrid; AUD 115,000 - 163,000; Full time

Benefits offered by this job

Competitive salary

Pension plan

Professional development opportunities

Networking opportunities

Vibrant Cambridge office

Job summary

CommonAI CIC invites a driven engineer to design and run reproducible experiments across hardware and software to illuminate AI infrastructure behavior. You will build measurement tools, collect traces and profiles, and help calibrate simulator models for a growing lab focused on scaling inference.

You will explore bottlenecks, performance variation and energy measurement across components, collaborating with workload, simulation and uncertainty specialists.

Qualifications

• Depth in accelerator profiling, inference runtimes, memory systems or HPC workloads.
• Experience with LLM serving and profiling tools like Nsight or PyTorch Profiler is useful but not required.

Responsibilities

• Design and run reproducible experiments on hardware/software components and complete LLM serving systems.
• Develop reusable measurement tools; collect traces, profiles, and counters for latency, throughput, memory, power and energy.
• Investigate bottlenecks and performance variation across stack layers with colleagues.
• Ensure experimental controls, reproducibility, and calibrated simulator models.
• Collaborate with simulation and uncertainty specialists to select measurements that resolve unknowns.

Skills

Accelerator profiling

Inference runtimes

Memory systems

Distributed HPC

Networking

Power measurement

LLM serving

Tools

Nsight

PyTorch Profiler

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.