Compiler Engineer [Contractor]

Company Overview :

Flux Computing designs and manufactures optical processors to train and run inference on large AI models.

The Contract :

We’re seeking a highly experienced and motivated Compiler Engineer [contractor] for a 6 month engagement outside of IR-35 to work on our ML Backend and functional simulator for our next-generation OTPUs with our permanent team. Your mandate will be to deliver a production-ready ML backend (compiler, runtime, debugger) and functional simulator that can accurately run most large AI models.

This role demands deep expertise in C and C++ programming, ML framework internals, compiler construction, and optimisation techniques.

Key Deliverables :

• Implement Runtime : Build a runtime that seamlessly integrates with PyTorch, JAX, and TensorFlow (PJRT) for both training and inference execution patterns. The runtime must support asynchronous execution and multiple devices.
• Implement Compiler : Build a compiler that is extensible to future optimisation patterns across operation fusion, layout optimisation, tiling, and scheduling.
• Implement Debugger & Diagnostics : Support optional runtime assertions and compile-time dumps, tensorboard timelines, and JAX I / O callbacks.
• Implement Functional Simulator : Build a functional simulator that mocks our kernel-space driver, allowing the software team to lower operations ahead of hardware teams.

Skills & Experience :