Senior / Staff Software Engineer (AI / Compiler)
Southampton, England, United Kingdom
JR United Kingdom
bottlenecks across the stack, from 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, ONNX, OpenXLA) to better support … focus on real-time or 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 Experience scaling AI workloads across More ❯
Posted: