Here is a very simple toy example for a mixture density network and SLOGAN 6.
There are also 8 Gaussians and friends, in pytorch eg in [thus improved WGAN notebook].(https://github.com/t-vi/pytorch-tvmisc/blob/master/wasserstein-distance/Improved_Training_of_Wasserstein_GAN.ipynb 7)
It is a neat topic you’ll enjoy exploring it.
Best regards
Thomas