jeremiecoullon/SGMCMCJax jeremiecoullon/SGMCMCJax

  • Stars
    star
    93
  • Rank 357,800 (Top 8 %)
  • Language
    Python
  • License
    Apache License 2.0
  • Created about 3 years ago
  • Updated 11 months ago
Twitter logo Share LinkedIn logo Share Facebook logo Share
jeremiecoullon/SGMCMCJax
github

jeremiecoullon

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Lightweight library of stochastic gradient MCMC algorithms written in JAX.

More Repositories

1

jax-tqdm

Add a tqdm progress bar to your JAX scans and loops.
Python
63
star
2

jax_MCMC_blog_post

Code to reproduce Jax samplers from blog post
Python
9
star
3

SGMCMC_bandit_tuning

Code for paper: Stochastic Gradient MCMC with Multi-Armed Bandit Tuning: https://arxiv.org/abs/2105.13059
Jupyter Notebook
4
star
4

knowitwall

Old version of KiW. Current version is here: https://github.com/jeremiecoullon/knowitwall-django
HTML
2
star
5

natural_numbers_post

Code for the "Implementing natural numbers in OCaml" blog post
OCaml
2
star
6

functional_ensemble_sampler

Code for paper: https://arxiv.org/abs/2010.15181
Python
2
star
7

BIP_LWR-paper

Jupyter Notebook
1
star