2 of 2 TensorBoard Jobs in the UK

Build a compiler that is extensible to future optimisation patterns across operation fusion, layout optimisation, tiling, and scheduling. Implement Debugger & Diagnostics: Support optional runtime assertions and compile-time dumps, tensorboard timelines, and JAX I/O callbacks. Implement Functional Simulator: Build a functional simulator that mocks our kernel-space driver, allowing the software team to lower operations ahead of hardware More ❯

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 ❯