mcgrady20150318 / pysgmcmc

Stochastic Gradient MCMC Methods for Tensorflow

PYSGMCMC

PYSGMCMC is a Python framework for Bayesian Deep Learning that focuses on Stochastic Gradient Markov Chain Monte Carlo methods.

Features

• Complex samplers as black boxes, computing the next sample with corresponding costs of any MCMC sampler is as easy as:

sample, cost = next(sampler)

• Based on tensorflow that provides:

  • efficient numerical computation via data flow graphs
  • flexible computation environments (CPU/GPU support, desktop/server/mobile device support)
  • Linear algebra operations

Install

The quick way:

pip3 install git+https://github.com/MFreidank/pysgmcmc

Documentation

Our documentation can be found at http://pysgmcmc.readthedocs.io/en/latest/.