A PyTorch implementation of InfoVAE: Balancing Learning and Inference in Variational Autoencoders by Zhao et al.