I'd be curious to read up on the parts of XLA design that were inspired by Halide if you have any links. I haven't familiarized myself with XLA/MLIR source code much, but I was under the impression it took a more conventional approach to optimizing compilers (pattern matching rather than generic separation of schedule and compute).