This is a full implementation of a simple VAE written entirely in Numpy (and Cupy). The code runs very slow on CPU so using a GPU with Cupy is recommended.
Original Paper: Auto-Encoding Variational Bayes, Diederik P Kingma, Max Welling
Used preprocessing code and inspired by @shinseung428's Vanilla GAN.
- Learns the MNIST Dataset
- Xavier Initialization
- Adam Optimizer
- Numpy
- PIL (visualize results)
- Cupy (Optional)
K / @kwj2104