HSVI is a small package for applying Stochastic Variational Inference to probabilistic models. It supports both tensorflow (v1.15) and pytorch (v1.9.0).
Install required packages:
pip install -r requirements.txt
Install HSVI on local machine:
python setup.py install
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.