About the Opportunity

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 spe...