Senior / Staff Software Engineer (AI / Compiler)

Flux Computing; London; On-site; GBP 145,000 - 180,000; Full time

Job summary

A leading company in AI technology is seeking Senior / Staff Software Engineers to develop innovative computing infrastructure for Optical Tensor Processing Units. This role involves designing systems for low-latency AI workloads and optimizing performance across distributed clusters. Ideal candidates will have extensive experience in high-performance computing and a passion for cutting-edge AI technologies.

Benefits

Stock options

Comprehensive healthcare insurance

Private use of 3D printer

Incentive for living within 20 minutes commute

Qualifications

• 5+ years of experience in HPC, HFT, or AI infrastructure.
• Deep understanding of distributed systems and real-time data processing.
• Hands-on experience with ML compilers and performance-sensitive applications.

Responsibilities

• Design and build high-performance systems for AI/ML workloads.
• Optimize for ultra-low latency and real-time inference.
• Collaborate with hardware teams to leverage OTPU architecture.

Skills

C++

Python

Distributed Systems

Performance Tuning

Debugging

Education

Degree in Computer Science, Engineering, Mathematics

Tools

LLVM

MLIR

PyTorch

ONNX

OpenXLA

Job description

Senior / Staff Software Engineer (AI / Compiler)

Senior / Staff Software Engineer (AI / Compiler)

3 weeks ago Be among the first 25 applicants

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 Optical Tensor Processing Units (OTPUs). This isn’t just about scaling models—it’s about rethinking how AI workloads are executed at speed and scale.

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 of the stack—from model representation and execution to data movement and scheduling. Whether it’s through compiler techniques, systems-level tuning, or custom runtime design, you’ll play a critical role in shaping the performance layer of our AI platform. This is a role for engineers who think in microseconds, not just model accuracy. If you’ve worked in HFT, large-scale scientific compute, or AI infrastructure at serious scale, we’d love to talk.

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 critical systems as needed
• Identify and resolve performance bottlenecks across the stack, from model execution and scheduling to hardware-level 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 ensure the software stack fully leverages the capabilities of our OTPU architecture
• Extend ML frameworks (e.g. PyTorch, ONNX, OpenXLA) to better support performance-critical inference paths
• Lead design reviews, mentor engineers, and promote best practices in HPC and performance engineering
• Stay on the frontier of new developments in AI infrastructure, compute systems, and compiler tooling

Skills & Experience

• 5+ years of experience building performance-critical systems in HPC, HFT, large-scale simulation, or AI infrastructure
• Deep understanding of distributed systems, with a focus on real-time or near real-time data processing
• Strong programming skills in C++ and Python, especially for performance-sensitive applications
• Hands-on experience with ML compilers (e.g. LLVM, MLIR), and knowledge of runtime and scheduling optimisations
• Practical knowledge of ML frameworks like PyTorch, ONNX, or OpenXLA, and how to optimise their execution
• Experience scaling AI workloads across clusters or custom infrastructure—not just deploying on standard cloud setups
• Strong debugging, profiling, and performance-tuning skills across the stack
• Degree in Computer Science, Engineering, Mathematics, or a related field

Details

• Competitive salary ranging from £145k+, depending on experience.
• 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.
• Flux hires candidates within a 45-minute commute of our office—offering an extra £24k/year incentive if you choose to live within 20 minutes. The <45-minute commute is a firm requirement.
• Comprehensive healthcare insurance.
• Private use of our 3D printer.

If you’re passionate about compilers, high-performance computing, and redefining what’s possible in AI, we’d love to talk. Apply now to join Flux and help shape the future of optical computing.

Seniority level

• Seniority level

  Mid-Senior level

Employment type

• Employment type

  Full-time

Job function

• Job function

  Engineering and Information Technology

• Industries

  Semiconductor Manufacturing

Referrals increase your chances of interviewing at Flux Computing by 2x

Get notified about new Senior Staff Software Engineer jobs in London, England, United Kingdom.

London, England, United Kingdom 1 month ago

Codified Controls - Senior Full-Stack Software Engineer - VP

London, England, United Kingdom 2 weeks ago

Lead Software Engineer - Transfer Agency team.

London, England, United Kingdom 1 week ago

Lead Software Engineer, Legal & Compliance

London, England, United Kingdom 20 hours ago

London, England, United Kingdom 3 days ago

London, England, United Kingdom 2 weeks ago

London, England, United Kingdom 1 week ago

London, England, United Kingdom 1 week ago

Lead Software Engineer - Electronic Trading (Vice President)

City Of London, England, United Kingdom 1 day ago

Principal Generative AI Software Engineer - SVP

London, England, United Kingdom 6 days ago

Lead Software Engineer - EU Natural Gas & Power

London, England, United Kingdom 2 months ago

Senior C++ Engineer (Mandarin Speaker) - London / Remote from the UK

London, England, United Kingdom 1 month ago

London, England, United Kingdom 5 days ago

London, England, United Kingdom 1 week ago

London, England, United Kingdom 1 week ago

London, England, United Kingdom 1 day ago

London, England, United Kingdom 4 weeks ago

London, England, United Kingdom 1 month ago

Senior Backend Software Engineer - London

London, England, United Kingdom 2 weeks ago

London, England, United Kingdom 3 months ago

London, England, United Kingdom 3 hours ago

London, England, United Kingdom 1 week ago

London, England, United Kingdom 1 week ago

London, England, United Kingdom 1 day ago

London, England, United Kingdom 2 weeks ago

London, England, United Kingdom 2 days ago

Senior Software Engineer, Backend (Relocation to the UK)

London, England, United Kingdom 3 months ago

London, England, United Kingdom 3 weeks ago

London, England, United Kingdom 2 weeks ago

London, England, United Kingdom 8 months ago

London, England, United Kingdom 2 weeks ago

London, England, United Kingdom 5 days ago

London, England, United Kingdom 1 week ago

London, England, United Kingdom 8 months ago

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.