MLIR: accelerating AI with open-source infrastructure