Performance Modeling Engineer

ZipRecruiter; London; On-site; GBP 60,000 - 90,000; Full time

Job summary

Join a leading AI company in London as a Performance Modeling Engineer, where you'll create and own analytical models that guide software and hardware evolution. With a focus on data-driven decision-making, you'll work in a dynamic environment and directly influence the optical compute system's design and functionality.

Benefits

Competitive salary and stock options

Comprehensive healthcare insurance

25 days PTO plus bank holidays

Private access to in-house 3D printer

Qualifications

• 3+ years building performance or power models.
• Strong coding ability in C++ and Python.
• Familiarity with machine-learning workloads.

Responsibilities

• Design, build, and maintain functional and performance simulators.
• Conduct workload analysis and identify bottlenecks.
• Develop tooling for statistical analysis and visualization.

Skills

Performance modeling

Power modeling

Data visualization

Communication

Education

Bachelor’s or higher in EE, CS, Physics, Applied Maths

Tools

Python

C++

gem5

SystemC

Job description

Job Description

Company Overview

Flux Computing designs and manufactures optical processors to train and run inference on large AI models. Join us in London to be part of a highly motivated and skilled team that thrives on delivering impact and innovation at speed.

The Role

We’re searching for a Performance Modeling 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 “what-if” design spaces, and deliver insights that directly influence our software, hardware, and optical roadmaps. This role 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

• 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 tradeoffs 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

• 3+ 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 but not required.
• Personal or open-source projects in simulators, ML kernels, or performance analysis are a significant plus.

Compensation & Benefits

• Competitive salary and stock options in a rapidly growing AI company.
• Based in our new 5,000 sq. ft. office in the AI hub of Kings Cross, London.
• To foster collaboration in our high-growth environment, we require all employees to work from our London HQ and live within a 45-minute commute. We offer an extra £24,000/year incentive for those living within 20 minutes.
• Comprehensive healthcare insurance.
• 25 days PTO policy plus bank holidays.
• Private access to our in-house 3D printer.

If you are passionate about pushing the boundaries of what's possible in AI and thrive in a high-energy, fast-paced environment, we want to hear from you. Apply now to join Flux and be a key player in shaping the future of computing.