8 Supplementary Material

Attached IPython notebooks were tested to work as expected in Colab. On replacing the Gaussian prior with a learned density in normalizing flows. As mentioned in the main paper, FFJORD LRMF performed on par with Real NVP version. The dynamics can be found in the Figure 10. Rightmost column shows LRMF convergence.