Senior / Staff Software Engineer (AI / Compiler)

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

Job summary

A leading company in AI accelerators is seeking Senior and Staff Software Engineers in Crawley, West Sussex. This exciting role requires expertise in designing distributed AI systems with a focus on performance and optimization. Candidates will work closely with hardware teams and play a pivotal role in revolutionizing AI workload execution. Attractive salary and stock options available, with incentives for local candidates.

Benefits

Stock options in a growing AI company

Extra £24k/year for living within 20 minutes

Qualifications

• 5+ years in performance-critical systems (HPC, HFT, AI infrastructure).
• Deep understanding of distributed systems and real-time data processing.

Responsibilities

• Design and build high-performance distributed AI/ML systems.
• Optimize for ultra-low latency and real-time inference.
• Identify and resolve performance bottlenecks.

Skills

C++

Python

Debugging

Performance tuning

Distributed systems

Real-time data processing

ML compilers

Optimization techniques

Education

Degree in CS, Engineering, Math

Job description

Social network you want to login/join with:

Senior / Staff Software Engineer (AI / Compiler), Crawley, West Sussex

Company Overview

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

The role

We’re hiring Senior and Staff Software Engineers to build high-performance computing infrastructure for OTPUs. This role involves rethinking AI workload execution at speed and scale.

You’ll lead design and implementation of distributed, low-latency inference systems, working closely with hardware and ML teams to optimize the stack—from model representation to data movement. Experience in microsecond-level performance is essential. Ideal candidates have backgrounds in HFT, scientific compute, or large-scale AI infrastructure.

Responsibilities

• Design and build high-performance distributed AI/ML systems
• Optimize for ultra-low latency and real-time inference, profiling and tuning as needed
• Identify and resolve performance bottlenecks across the stack
• Collaborate with compiler engineers to improve code generation and execution
• Work with hardware teams to leverage OTPU capabilities
• Extend ML frameworks (PyTorch, ONNX, OpenXLA) for performance
• Lead reviews, mentor engineers, promote HPC best practices
• Stay updated on AI infrastructure and compiler tooling

Skills & Experience

• 5+ years in performance-critical systems (HPC, HFT, AI infrastructure)
• Deep understanding of distributed systems and real-time data processing
• Strong C++ and Python skills
• Experience with ML compilers like LLVM, MLIR
• Knowledge of ML frameworks and optimization techniques
• Experience scaling AI workloads on custom infrastructure
• Debugging, profiling, performance tuning skills
• Degree in CS, Engineering, Math, or related field

Details

• Salary from £145k+, depending on experience
• Stock options in a growing AI company
• Located in Kings Cross, London
• Candidates within 45-minute commute preferred; extra £24k/year incentive for living within 20 minutes