Senior AI Performance Modeller

Flux Computing; City Of London; On-site; GBP 60,000 - 80,000; Full time

Job summary

A leading technology firm in London is searching for a Senior Performance Modelling Engineer to create analytical models that influence software and hardware decisions. You will collaborate closely with various teams to ensure models reflect reality, handle complex data, and significantly guide the company's technology roadmaps. A strong background in programming with C++ and Python, as well as experience in performance modeling, is essential for this role. The position includes benefits like competitive salary and full healthcare coverage.

Benefits

Competitive salary

Stock options

Full healthcare and dental cover

Monthly off-site team socials

25 days of paid holiday

Cycle2work scheme

Qualifications

• 5+ years building performance or power models for CPUs, GPUs, ASICs, or accelerators.
• Strong coding ability in C++ and Python; experience with discrete-event or cycle-accurate simulators.
• Solid grasp of computer-architecture fundamentals: memory systems, interconnects, queuing theory.
• Familiarity with machine-learning workloads and common frameworks.

Responsibilities

• Own and deliver projects on the team’s roadmap.
• Work with hardware, compiler and ML framework teams.
• Design, build, and maintain a functional simulator of the OPTU subsystem.
• Identify throughput, latency and utilisation hot-spots.

Skills

C++

Python

Performance modeling

Data visualization

Machine learning frameworks

Education

Bachelor’s+ in EE, CS, Physics, Applied Maths or related

Tools

gem5

SystemC

Job description

The Role

We’re searching for a Senior Performance Modelling Engineer to create and own the analytical and simulation models that steer OTPU architecture and software evolution. You will build functional simulators as well as high‑fidelity, cycle‑accurate models of our optical compute system. This role is critical to explore "whatif" design spaces, and deliver insights that directly influence our software, hardware, and optical roadmaps. It sits at the crossroads of hardware architecture, software tooling and machine‑learning workload analysis, perfect for an engineer who loves data‑driven decision‑making and fast iteration.

Responsibilities

• Project Ownership: Own and deliver projects on the team’s roadmap that unlock key high‑impact technical and business milestones that drive the success of Flux.
• Collaboration: Work shoulder‑to‑shoulder with hardware, compiler and ML framework teams to ensure models reflect reality and reality meets performance goals.
• Functional Simulator: Design, build, and maintain a functional simulator of the OPTU subsystem and full pipeline.
• Performance Simulator: Design and maintain architectural & cycle‑accurate models of the OPTU subsystems and pipeline. Identify throughput, latency and utilisation hot‑spots; propose architectural, or scheduling fixes.
• Workload Analysis & Bottleneck Hunting: Instrument benchmarks (LLMs, diffusion, graph workloads) to collect detailed traces.
• Design‑Space Exploration: Run massive parameter sweeps with your functional and to understand trade‑offs and guide the software, hardware, and optical teams. Package results into clear, quantitative analyses and design recommendations.
• Tooling & Automation: Develop Python/C++ tooling for trace parsing, statistical analysis and visualisation. Integrate models into CI so that every RTL commit gets a performance smoke test.

Skills & Experience

• 5+ years building performance or power models for CPUs, GPUs, ASICs, or accelerators.
• Strong coding ability in C++ and Python; experience with discrete‑event or cycle‑accurate simulators (e.g., gem5, SystemC, custom in‑house).
• Solid grasp of computer‑architecture fundamentals: memory systems, interconnects, queuing theory, Amdahl/Gustafson analysis.
• Familiarity with machine‑learning workloads and common frameworks (PyTorch, TensorFlow, JAX).
• Comfort reading RTL or schematics and discussing micro‑architectural trade‑offs with hardware designers.
• Excellent data‑visualisation and communication skills: able to turn millions of simulation samples into one decisive slide.
• Bachelor’s+ in EE, CS, Physics, Applied Maths or related; advanced degree preferred but not required.
• Personal or open‑source projects in simulators, ML kernels, or performance analysis are a significant plus.

Compensation & Benefits

• Starting salary £174,000+ depending on experience, skills, and location.
• Competitive stock options, you’ll own a piece of the journey.
• Work from our HQ in King’s Cross, right in the middle of London’s buzzing AI scene.
• Live within 45 minutes of the office? Perfect. Live within 20 minutes? We’ll add an extra location bonus to your salary.
• We offer visa sponsorship and full relocation support (UK and abroad), through a dedicated third‑party provider who are on hand to make your move to London as seamless as possible.
• Full BUPA healthcare and dental cover, medical history disregarded.
• High‑spec tech for everyone – M4 Macs as standard, M4 Pros for Engineers.
• Sony noise‑cancelling headphones and ergonomic setups to keep you comfortable and focused.
• Personal company card to spend on tools that help you do your job – like ChatGPT Pro or anything else that boosts your workflow.
• Healthy, chef‑cooked dinners in the office every night, with something for every diet and tastebud.
• Monthly off‑site team socials.
• 25 days of paid holiday, plus all the UK bank holidays.
• Access to our in‑house 3D printer for personal or work projects.
• Cycle2work scheme.
• Need a caffeine fix? We’ve got you covered with a tab at our favourite local coffee shop.
• We offer a pension plan and salary sacrifice options.

Due to U.S. export control regulations, candidates' eligibility to work at Flux depends on their most recent citizenship or permanent residency status. We are generally unable to consider applicants whose most recent citizenship or permanent residence is in certain restricted countries (currently including Iran, North Korea, Syria, Cuba, Russia, Belarus, China, Hong Kong, Macau, and Venezuela). Applicants who have subsequently obtained citizenship or permanent residency in another country not subject to these restrictions may still be eligible.

We do not accept unsolicited CVs from recruitment agencies, will not be liable for any fees, and prohibit unauthorised use of our company name in recruitment activities.