Writing an MCMC sampler in JAX by Jeremie Coullon
Tutorial on the different ways to write an MCMC sampler in JAX along with speed benchmarks.
- from
- JAX
- added
- 2026-10-10
- likes
- 0
similar
-
-
-
Jax in Action manning.com
A hands-on guide to using JAX for deep learning and other mathematically-intensive applications.
-
-
-
Writing a Training Loop in JAX + FLAX by Saurav Maheshkar and Soumik Rakshit wandb.ai
A tutorial on writing a simple end-to-end training and evaluation pipeline in JAX, Flax and Optax.
JAX › Tutorials and Blog Posts: “Tutorial on the different ways to write an MCMC sampler in JAX along with speed benchmarks.”