JTN-VAE model implementation

[gcc17]

Hi! Thanks a lot for your efforts in integrating these models.
In the README, you show the result of JTN-VAE. However, it seems that there has no real implementation for JTN-VAE. Could you show me a way to use JTN-VAE? Thanks!