Senior / Staff Software Engineer (AI / Compiler)
northampton, midlands, United Kingdom
Flux Computing
model execution and scheduling to hardware-level constraints Collaborate with compiler engineers to improve code generation, execution paths, and memory layouts using tools like LLVM or MLIR Work with hardware teams to ensure the software stack fully leverages the capabilities of our OTPU architecture Extend ML frameworks (e.g. PyTorch … near real-time data processing Strong programming skills in C++ and Python, especially for performance-sensitive applications Hands-on experience with ML compilers (e.g. LLVM, MLIR), and knowledge of runtime and scheduling optimisations Practical knowledge of ML frameworks like PyTorch, ONNX, or OpenXLA, and how to optimise their execution More ❯
Posted: