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