ML Complier Engineer

Job Description : Compiler Engineer [Contractor]

Company Overview

designs and manufactures optical processors built to train and run inference on large-scale AI models.

Contract Overview

We are seeking an experienced and highly motivated Compiler Engineer (Contractor) . You’ll collaborate with our in-house engineering team to deliver a production-ready ML backend —including compiler, runtime, debugger, and a functional simulator—for our next-generation Optical Tensor Processing Units (OTPUs).

This role requires deep technical expertise in compiler construction, runtime systems, C / C++ programming, and ML framework internals.

Key Responsibilities

Compiler Development

Design and build a compiler extensible to future optimisation patterns such as op fusion, layout optimisation, tiling, and scheduling.

Debugger & Diagnostics

Implement optional runtime assertions, compile-time dump mechanisms, TensorBoard timelines, and JAX I / O callback support.

Functional Simulator

Develop a simulator that mimics our kernel-space driver, enabling the software team to lower operations ahead of hardware availability.

Required Skills & Experience

5+ years of professional experience in C / C++ software engineering

Strong background in compilers , runtime systems , and low-level optimisations

Deep familiarity with ML frameworks (e.g., PyTorch, JAX, TensorFlow) and their execution models

Experience with high-performance computing or hardware-software co-design

Strong problem-solving skills with a creative and pragmatic mindset

Comfortable operating in fast-paced , ambiguous environments

Bachelor’s degree (or higher) in Computer Science, Electrical Engineering, Mathematics, or a related field

Excellent communication and collaboration skills

Contract Details

Length : 3+ months

IR35 Status : Outside IR35

Location : Remote (UK or EU preferred)

Start Date : ASAP