Senior / Staff Software Engineer (AI / Compiler)

JR United Kingdom; Leeds; On-site; GBP 145,000 - 180,000; Full time

Job summary

A leading AI company is seeking Senior and Staff Software Engineers in Leeds. The role focuses on optimizing the performance of AI workloads across distributed computing environments, requiring a strong background in C++, Python, and experience in high-performance computing. Join a dynamic and innovative team pushing the boundaries of AI technology in a supportive environment.

Benefits

Stock options

Competitive salary

Incentives for nearby commuters

Qualifications

• 5+ years building performance-critical systems.
• Deep experience in distributed systems and real-time processing.
• Strong skills in HPC, HFT, or scaling AI workloads.

Responsibilities

• Design and build high-performance systems for AI workloads.
• Optimize for low latency and real-time inference at scale.
• Collaborate with compiler engineers and hardware teams.

Skills

C++

Python

Real-time data processing

Performance tuning

Debugging

Education

Degree in Computer Science

Engineering

Mathematics

Tools

LLVM

MLIR

PyTorch

ONNX

OpenXLA

Job description

Social network you want to login/join with:

Senior / Staff Software Engineer (AI / Compiler), Leeds, West Yorkshire

Company Overview

Flux is pioneering a new class of AI accelerators called Optical Tensor Processing Units (OTPUs). We’ve already developed functioning prototypes and are now scaling our operations in London. Our work environment rewards innovation, speed, and bold thinking.

The role

We’re hiring Senior and Staff Software Engineers to build the high-performance computing infrastructure that powers our OTPUs. This involves rethinking how AI workloads are executed at speed and scale, not just scaling models.

You’ll lead the design and implementation of software systems that run distributed, low-latency inference across clusters. You’ll work closely with hardware and ML teams to optimise every layer—from model representation and execution to data movement and scheduling. Whether through compiler techniques, systems tuning, or custom runtime design, you’ll shape the performance layer of our AI platform. This role is suited for engineers who think in microseconds, not just model accuracy. Experience in HFT, large-scale scientific compute, or AI infrastructure at scale is desirable.

Responsibilities

• Design and build high-performance systems for running AI/ML workloads across distributed compute clusters
• Optimise for ultra-low latency and real-time inference at scale—profiling, tuning, and rewriting systems as needed
• Identify and resolve performance bottlenecks across the stack, from model execution to hardware constraints
• Collaborate with compiler engineers to improve code generation, execution paths, and memory layouts using tools like LLVM or MLIR
• Work with hardware teams to leverage OTPU architecture capabilities
• Extend ML frameworks (e.g., PyTorch, ONNX, OpenXLA) for performance-critical inference paths
• Lead design reviews, mentor engineers, and promote best practices in HPC and performance engineering
• Stay updated on developments in AI infrastructure, compute systems, and compiler tooling

Skills & Experience

• 5+ years building performance-critical systems in HPC, HFT, large-scale simulation, or AI infrastructure
• Deep understanding of distributed systems, especially real-time data processing
• Strong programming skills in C++ and Python for performance-sensitive applications
• Experience with ML compilers (LLVM, MLIR), runtime, and scheduling optimisations
• Knowledge of ML frameworks like PyTorch, ONNX, OpenXLA, and their optimisation
• Experience scaling AI workloads across clusters or custom infrastructure
• Strong debugging, profiling, and performance tuning skills
• Degree in Computer Science, Engineering, Mathematics, or related field

Details

• Competitive salary from £145k+, depending on experience
• Stock options in a rapidly growing AI company
• Based in our new 5,000 sq. ft. office in Kings Cross, London
• Flux hires candidates within a 45-minute commute, with an extra £24k/year incentive for those living within 20 minutes