Job Description
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...