Could you please release the code of training the joint prior from a scratch?

[Phonicavi]

Thank you for sharing this awesome work!
I was trying to reproduce the normalizing flow used in the joint prior parts, and I noticed that the network structure (e.g. the PReLU implementation) might be different from the RealNVP, or Low-capacity version of normalizing flows mentioned in https://arxiv.org/abs/2003.10350
Could you please release the method or the code for training this network?

[fantaosha]

Sorry for the late reply. We're working on a new joint priors and will release the training code once it is done.

I will keep this issue open until the code is released.

[vinkle]

Thank you so much for your excellent work. It would be great to let me know any timeline when you will release the training code for joints prior.

[aashusri]

Any update on this?