The variational infomax autoencoder