A library for developing deep generative models
Pixyz is a high-level deep generative modeling library, based on PyTorch. It is developed with a focus on enabling easy implementation of various deep generative models.
Pixyz enables users to implement deep generative models (VAE, GAN, etc.) intuitively, as if writing the formulas on paper.
Pixyz APIs provide independent definitions for deep neural networks, probablistic models and optimizers. This enables users to implement even for complex models concisely.
pip install pixyz
or
docker pull kenoharada/pixyz:v0.3.0_python_3.7.7_pytorch_1.6.0_cuda_10.1