Senior Compiler Engineer: ML Backend for AI Hardware

Olix Computing Ltd.; Greater London; Hybrid; GBP 120,000 - 180,000; Full time

Benefits offered by this job

Stock options

Living-Local Bonus

Employer-contributed retirement plan

Competitive salary

Job summary

Olix Computing Ltd. in London invites a Senior Compiler Engineer to help build the ML backend for the DX-1 accelerator. You will connect PyTorch, TensorFlow, JAX and MXNet to our low-level kernel drivers on custom hardware, pushing performance across large AI models.

You will own projects, design and implement compiler, runtime, and debugger components, and mentor peers while tracking industry trends in GPUs and AI accelerators. A strong C/C++ background and ML framework knowledge are essential.

Qualifications

• 5+ years of software engineering experience with focus on C/C++ programming.
• Deep expertise in ML framework internals, compilers, low-level programming and optimisation techniques.
• Deep expertise with optimizing TensorFlow, PyTorch or JAX models.
• Experience with LLVM, OpenXLA/XLA, MLIR, TVM toolchains.
• Practical experience applying ML in HPC contexts.
• Strong problem-solving and critical thinking abilities.
• Experience in high-pace, dynamic environments.
• Excellent teamwork and cross-functional collaboration abilities.
• Bachelor's degree in computer science, electrical engineering, telecoms engineering, mathematics, or related field.

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

C/C++ programming

ML framework internals

TensorFlow / PyTorch / JAX

Compiler backends

Low-level programming

Education

Bachelor's degree in CS or related field

Tools

LLVM

OpenXLA/XLA

MLIR

TVM

Job description

Olix Computing Ltd. in London invites a Senior Compiler Engineer to help build the ML backend for the DX-1 accelerator. You will connect PyTorch, TensorFlow, JAX and MXNet to our low-level kernel drivers on custom hardware, pushing performance across large AI models.

You will own projects, design and implement compiler, runtime, and debugger components, and mentor peers while tracking industry trends in GPUs and AI accelerators. A strong C/C++ background and ML framework knowledge are essential.