Sounds very interesting. Could you explain a bit more what you did and why it was easier with MLIR?