Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
3 contributors

Users who have contributed to this file

@thjashin @csy530216 @Li-Scottech
99 lines (72 sloc) 2.77 KB

Welcome to ZhuSuan

_static/images/index.png

ZhuSuan is a python probabilistic programming library for Bayesian deep learning, which conjoins the complimentary advantages of Bayesian methods and deep learning. ZhuSuan is built upon Tensorflow. Unlike existing deep learning libraries, which are mainly designed for deterministic neural networks and supervised tasks, ZhuSuan provides deep learning style primitives and algorithms for building probabilistic models and applying Bayesian inference. The supported inference algorithms include:

  • Variational inference with programmable variational posteriors, various objectives and advanced gradient estimators (SGVB, REINFORCE, VIMCO, etc.).
  • Importance sampling for learning and evaluating models, with programmable proposals.
  • Hamiltonian Monte Carlo (HMC) with parallel chains, and optional automatic parameter tuning.
.. toctree::
   :maxdepth: 2


Installation

ZhuSuan is still under development. Before the first stable release (1.0), please clone the GitHub repository and run

pip install .

in the main directory. This will install ZhuSuan and its dependencies automatically. ZhuSuan also requires Tensorflow version 1.13.0 or later. Because users should choose whether to install the cpu or gpu version of Tensorflow, we do not include it in the dependencies. See Installing Tensorflow.

If you are developing ZhuSuan, you may want to install in an "editable" or "develop" mode. Please refer to the Contributing section in README.

After installation, open your python console and type:

>>> import zhusuan as zs

If no error occurs, you've successfully installed ZhuSuan.

.. toctree::
   :maxdepth: 1
   :caption: Tutorials

   Tutorial slides <https://docs.google.com/presentation/d/1Xqi-qFHciAdV9z1FHpGkUcHT-yugNVzwGX3MM74rMuM/edit?usp=sharing>
   tutorials/vae
   tutorials/concepts
   tutorials/bnn
   tutorials/lntm


.. toctree::
   :maxdepth: 1
   :caption: API Docs

   api/zhusuan.distributions
   api/zhusuan.framework
   api/zhusuan.variational
   api/zhusuan.hmc
   api/zhusuan.sgmcmc
   api/zhusuan.evaluation
   api/zhusuan.transform
   api/zhusuan.diagnostics
   api/zhusuan.utils
   api/zhusuan.legacy

.. toctree::
   :maxdepth: 1
   :caption: Community

   contributing



Indices and tables

You can’t perform that action at this time.