ML Systems Engineer (Compiler & Graph Optimization)

Compiler Optimization Engineer / Remote or on-site / Well-funded startup

Rare opportunity to join a well-funded start-up building a hardware-agnostic AI compiler that allows teams to deploy to any accelerator architecture from a single codebase.

We are looking for a core engineer to join the team behind our graph optimization layer. In this role, you will have a direct hand in shaping how the next generation of AI models scale across diverse hardware.

About the role:

• You'll design, implement, and maintain graph-level optimisation passes including operator fusion, layout propagation, tiling, dead code elimination, and constant folding
• You'll get the chance to define and evolve the intermediate representation (IR) to support new optimisation opportunities as ML model architectures advance
• You'll analyse real performance data to identify gaps and drive measurable improvements in throughp...