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 ❯
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 ❯
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 ❯
London, England, United Kingdom Hybrid / WFH Options
Better Placed Ltd - A Sunday Times Top 10 Employer!
to end testing Any AI-Specific Testing Tools Model testing: MLflow for tracking experiments A/B testing: Split or GrowthBook for model comparison Performance monitoring: Weights & Biases or TensorBoard API & Integration Testing REST API testing: Postman, REST-assured, or direct pytest with requests The absolute key to the role is experience in AI startups working with small teams of More ❯
