Let's implement deep generative models more easily

A library for developing deep generative models

Features

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.

Intuitive

Intuitive

Pixyz enables users to implement deep generative models (VAE, GAN, etc.) intuitively, as if writing the formulas on paper.

Concise

Concise

Pixyz APIs provide independent definitions for deep neural networks, probablistic models and optimizers. This enables users to implement even for complex models concisely.

Various

Documented

We have various exapmles implemented in Pixyz. For basic models please visit here. For complicated models please visit here.

Quick start

pip install pixyz

or

docker pull kenoharada/pixyz:v0.3.0_python_3.7.7_pytorch_1.6.0_cuda_10.1


Contact

GitHub

Any requests, issues and pull requests are welcome.