Our INN architecture contains 8 blocks. Each block consists of a soft permutation of channels [1]}, an affine coupling layer [2]}, and an ActNorm layer [3]}. In the affine coupling layer, we model the sub-network as an MLP with one hidden layer, where the hidden dimension is 256 and the non-linear activation is LeakyReLU\(_{0.1}\) .
