VAE with a VampPrior
Human-level concept learning through probabilistic program induction