SGMCMCJax
stable

How to use SGMCMCJax

  • Overview
  • Build a sampler function
  • Build a transition kernel
  • Build a diffusion solver

Examples

  • Gaussian posterior
  • Logistic Regression
  • Bayesian Neural Network
  • Flax CNN
  • Sample from a PyMC model using SGMCMCJax

Samplers

  • SGMCMC samplers

Add a sampler to SGMCMCJax

  • Add a sampler
  • Special cases

FAQ

  • Frequently asked questions

API

  • Kernels
  • Diffusions
  • Samplers
  • Gradient estimation
  • Optimizer
  • Utility functions
  • Kernel Stein Discrepancy
SGMCMCJax
  • »
  • Overview: module code

All modules for which code is available

  • sgmcmcjax.diffusion_util
  • sgmcmcjax.diffusions
  • sgmcmcjax.gradient_estimation
  • sgmcmcjax.kernels
  • sgmcmcjax.ksd
  • sgmcmcjax.optimizer
  • sgmcmcjax.samplers
  • sgmcmcjax.util

© Copyright 2021, Jeremie Coullon. Revision d1327fae.

Built with Sphinx using a theme provided by Read the Docs.