Supervised VAE on dSprites, implemented with Pyro to allow counterfactual queries
Jupyter Notebook
CausalVAE
The jupyter notebook should run as is. It is configured to run on GPU and I would highly recommend for performance. You can switch to CPU at the top using the 'USE_CUDA' flag.