Senior Compiler Engineer

Flux Computing; Greater London; On-site; GBP 90,000 - 140,000; Full time

Benefits offered by this job

Equity options

Living-Local Bonus

Retirement benefits

Job summary

OLIX is seeking a Senior Compiler Engineer to join the team building the ML backend (compiler, run-time, debugger) for our next‑gen hardware. You will enable seamless PyTorch, TensorFlow, JAX integration, and drive performance optimisations on bespoke accelerators.

You will lead design and implementation of the compiler stack, collaborate with hardware and modelling teams, and mentor peers to maintain coding standards as the company scales its decode accelerator DX‑1 project.

Qualifications

• 5+ years of software engineering focusing on C/C++.
• Deep expertise in ML framework internals, compilers and optimisation.
• Strong experience with TensorFlow, PyTorch or JAX.
• Hands-on with LLVM, MLIR, OpenXLA/XLA, TVM toolchains.
• Proven ability to optimize large AI models in HPC contexts.
• Excellent problem-solving and communication skills.
• Demonstrated teamwork in fast-paced environments.
• Bachelor's degree in CS/EE or related field; strong personal projects.

Responsibilities

• Own and deliver compiler, runtime and debugger for ML backend.
• Design and implement support for PyTorch, TensorFlow, JAX on custom hardware.
• Apply advanced techniques to improve throughput and efficiency.
• Lead code reviews and mentor peers on performance engineering.
• Stay ahead of GPU accelerator trends and prototype innovations.

Skills

C/C++ programming

ML framework internals

TensorFlow / PyTorch / JAX

LLVM / MLIR / XLA

High-performance computing

Problem solving

Team collaboration

Personal projects

Education

Bachelor's degree in computer science or related field

Tools

LLVM

OpenXLA/XLA

MLIR

TVM

Job description

About OLIX

AI is growing faster than any technology in history and the explosion in demand has created a massive infrastructure gap; we can no longer build chips or power stations fast enough to keep up. The industry is still leaning on a ten-year-old hardware blueprint that has reached its limit. A new paradigm that is faster and more efficient will be the biggest economic opportunity of the next century and create the most important company of the next decade. The OLIX Decode Accelerator 1 (DX-1) is the first accelerator architected specifically for decode. Rack-scale co-design of logic, data movement, packaging, optics and interconnect enables a step change in system level performance.

The Role

We’re searching for Senior Compiler Engineer to join the team building the ML backend (compiler, run-time, and debugger) for our next-generation hardware that connect PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission will be to create seamless support for a broad ecosystem of large AI models, and ensure we are pushing the limits of their performance envelope by closely partnering with hardware and modelling teams to understand device trade-offs.

Responsibilities

• Project Ownership: Own and deliver projects on your team's roadmap that unlock key high-impact technical and business milestones that drive the success of Olix.

• Architect & Build: Design and implement our compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet on custom hardware.

• Optimise Performance: Apply advanced techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and maximise throughput.

• Mentor & Encourage Standards: Lead code reviews, coach peers, and uphold best practices in ML backend and performance engineering for your team.

• Guide Team Direction: Stay ahead of GPU, AI accelerator trends; propose and prototype innovations.

Skills & Experience

• 5+ years of experience in software engineering with a focus on C/C++ programming.

• Deep expertise in ML framework internals, compilers, low-level programming, and optimisation techniques.

• Deep expertise in optimising Tensorflow, PyTorch or JAX deep learning models.

• Deep expertise with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM.

• Practical experience applying machine learning in high-performance computing contexts.

• Strong problem-solving skills and the ability to think critically and creatively.

• Experience in high-pace, dynamic work environments.

• Excellent teamwork and communication skills, with the ability to collaborate effectively with cross‑functional teams.

• Bachelor's degree in computer science, electrical engineering, telecoms engineering, mathematics, or a related field.

• Personal projects are a key differentiating factor and hold more weight than other requirements.

Compensation & Equity

• Competitive Salary: Commensurate with your experience, skills, and location

• Equity & Ownership: Meaningful stock options. You’re not just joining the mission; you’re owning a piece of it

• Proximity Bonus: We value your time. To minimise your commute and maximise your life, we offer an annual Living‑Local Bonus if your residence is within 20 minutes of the office

• Retirement Benefits:Employer‑contributed retirement plans to help you build long‑term financial security.

Due to U.S. export control regulations, candidates’ eligibility to work at OLIX depends on their most recent citizenship or permanent residency status. We are generally unable to consider applicants whose most recent citizenship or permanent residence is in certain restricted countries (currently including Iran, North Korea, Syria, Cuba, Russia, Belarus, China, Hong Kong, Macau, and Venezuela). Applicants who have subsequently obtained citizenship or permanent residency in another country not subject to these restrictions may still be eligible.