A Pytorch implementation of Poisson Identifiable Variational Autoencoder (pi-VAE)
copied from cf-staging / pi-vae-pytorchPoisson Identifiable VAE (pi-VAE) is used to construct latent variable models of neural activity while simultaneously modeling the relation between the latent and task variables (non-neural variables, e.g. sensory, motor, and other externally observable states).