We are working with an ambitious AI infrastructure company building a new software stack that enables machine-learning workloads to run efficiently across diverse hardware targets.

They are looking for a Compiler Engineer to own the graph-optimisation layer of their AI compiler within a fast-paced start-up gaining a lot of commercial traction.

What you’ll do

- Design and implement graph-level optimisation passes for a heterogeneous AI compiler
- Work on fusion, layout optimisation, dead-code elimination, constant folding and algebraic simplification
- Evolve the compiler IR as model architectures and hardware targets develop
- Use performance analysis to identify optimisation opportunities and improve execution efficiency




- Partner closely with frontend and code-generation engineers on clean compiler interfaces and robust validation

What we’re looking for

- 4+ years of compiler engineering experience, particularly IR design or optimisation passes
- Solid C/C++
- Deep understanding of graph and tensor optimisation, including fusion, tiling and layout transformations
- Exposure to MLIR, XLA or comparable compiler frameworks would be valuable
- Familiarity with PyTorch, JAX, TensorRT, GPUs or AI accelerators is beneficial
- An interest in solving difficult performance problems at the intersection of ML models, compilers and hardware

📌 Compiler Engineer (Toronto)
🏢 Oho Group
📍 Toronto