Software Developer - ML/AI Compiler

Exciting opportunity to join an early-stage AI infrastructure company working at the intersection of large language models and compiler technology. You will be building core compiler components that translate deep learning models into optimised, hardware‑aware code working on a genuinely novel technical problem that sits at the frontier of how AI workloads get compiled and executed.

What You'll Do

• You'll get the chance to design and build compiler passes, runtime optimisations, and backend components targeting AI training workloads
• You'll have the opportunity to work directly with LLM‑based agent systems being used to automate code generation and kernel optimisation
• You'll analyse performance bottlenecks in training pipelines and apply optimisations at both framework and kernel levels
• You'll have the chance to collaborate closely with ML specialists and a dedicated code‑gen team to accelerate model training across diverse hard...