Hi Yilun,
Thanks for providing the source code. Would you be able to provide the pre-trained weights for the mixture of two Gaussian model? I understand this is a simple experiment, but I wish to reproduce some of the results in your paper and run an analysis during sampling.