Senior / Staff Software Engineer (AI / Compiler)

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

Job summary

A pioneering AI company is seeking Senior and Staff Software Engineers to develop high-performance computing infrastructure for Optical Tensor Processing Units. This role emphasizes optimizing AI workloads and requires expertise in performance-critical systems, C++, and Python. The position offers a competitive salary starting at £145k, stock options, and an additional incentive for local candidates.

Benefits

Stock options

Incentive for local commute

Qualifications

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

Responsibilities

• Design and build high-performance systems for AI/ML workloads across clusters.
• Collaborate with hardware teams to leverage OTPU architecture capabilities.

Skills

C++

Python

Distributed Systems

Real-time Data Processing

Debugging

Performance Tuning

Education

Degree in Computer Science, Engineering, Mathematics

Tools

ML Compilers (LLVM, MLIR)

ML Frameworks (PyTorch, ONNX, OpenXLA)

Job description

Social network you want to login/join with:

col-narrow-left

Client:

Location:

Job Category:

Other

-

EU work permit required:

Yes

col-narrow-right

Job Views:

5

Posted:

31.05.2025

Expiry Date:

15.07.2025

col-wide

Job Description:

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 role involves rethinking how AI workloads are executed at speed and scale, focusing on low-latency, high-performance systems.

You’ll lead the design and implementation of distributed software systems for inference across clusters, working closely with hardware and ML teams to optimize every layer of the stack, from model execution to data movement. This role requires engineers who think in microseconds and have experience in high-frequency trading, scientific computing, or large-scale AI infrastructure.

Responsibilities

• Design and build high-performance systems for AI/ML workloads across clusters
• 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 paths
• Work with hardware teams to leverage OTPU architecture capabilities
• Extend ML frameworks like PyTorch, ONNX, OpenXLA for performance
• Lead design reviews, mentor engineers, and promote best practices
• Stay updated on developments in AI infrastructure and compiler tooling

Skills & Experience

• 5+ years in performance-critical systems in HPC, HFT, simulation, or AI infrastructure
• Deep understanding of distributed and real-time data processing systems
• Strong skills in C++ and Python for performance applications
• Experience with ML compilers like LLVM, MLIR, and runtime optimizations
• Knowledge of ML frameworks like PyTorch, ONNX, OpenXLA
• Experience scaling AI workloads on custom infrastructure
• Strong debugging and performance tuning skills
• Degree in Computer Science, Engineering, Mathematics, or related field

Details

• Competitive salary starting at £145k+, depending on experience
• Stock options in a growing AI company
• Based in our new 5,000 sq. ft. office in Kings Cross, London
• Candidates within a 45-minute commute are preferred; an additional £24k/year incentive is offered for those living within 20 minutes