Software Engineer, AI Compiler

You'll work alongside hardware architects and researchers to co-design compiler strategies that unlock the full potential of our accelerator, building infrastructure that bridges the gap between ML model graphs and custom ISA primitives. Your compiler decisions will directly inform hardware features, and hardware capabilities will open new optimisation frontiers for your toolchain.If you want to architect a compiler stack from first principles, optimise ML workloads on new hardware, and see your decisions realised in silicon, this is the role.ResponsibilitiesWork across the full stack with software, systems, and hardware teams to ensure correctness, performance, and deployment readiness for real workloadsContribute to shaping the long-term compiler architecture and tooling strategy in a fast-moving startup environmentDesign and implement parts of the compiler stack targeting our novel AI accelerator, including front-end lowering, IR transformations, optimization passes, and backend code generationBuild and evolve MLIR/LLVM based infrastructure to support graph lowering, hardware-aware optimizations, and performance-centric code emissionCollaborate closely with hardware architects, microarchitects, and research teams to co-design compiler strategies that align with evolving ISA and hardware constraintsDevelop profiling and analysis tools to identify performance bottlenecks, validate ge...