Similarly to the model of Mishra-Sharma et al. [1]}, we employ masked autoregressive flows [2]} as the transformation flow with 8 transformations made of a 2-layer masked autoregressive MLP with a hidden layer size of 128. The model was implemented with nflows library [3]}. Model parameters are optimized by minimizing the following objective:
\(\mathcal {L}_{NFs}(x,y; \theta ) = - log \: p_K(x|y) = - log \: p(f^{-1}(x, y)) - \sum _{k=1}^K log \: \left| det \frac{dz_k}{dz_{k-1}}\right|\) 
