Masked Autoregressive Flow TF2.0 , how to add it with my VAE Network?

Question

does anyone already add MAF Layer to an existing model ? I\'m struggling into implement a solution based on MAF with TF-2.0 . I have a classic variational auto-encoder (VAE)