/hsvi

Primary LanguageJupyter Notebook

Hierarchical Stochastic Variational Inference (HSVI)

Introduction

HSVI is a small package for applying Stochastic Variational Inference to probabilistic models. It supports both tensorflow (v1.15) and pytorch (v1.9.0).

Installation

Install required packages:

pip install -r requirements.txt

Install HSVI on local machine:

python setup.py install

User Guide

Please find full details in the User Guide Page, which includes API introduction, several example models (BNN, VAE, and a probabilistic model) for both tensorflow and pytorch versions.