ML Complier Engineer
London, United Kingdom
La Fosse
Design and build a compiler extensible to future optimisation patterns such as op fusion, layout optimisation, tiling, and scheduling. Debugger & Diagnostics Implement optional runtime assertions, compile-time dump mechanisms, TensorBoard timelines, and JAX I/O callback support. Functional Simulator Develop a simulator that mimics our kernel-space driver, enabling the software team to lower operations ahead of hardware availability. More ❯
Employment Type: Contract
Posted: