How to add a progress bar to JAX scans and loops by Jeremie Coullon

jeremiecoullon.comsite

Tutorial on how to add a progress bar to compiled loops in JAX using the host_callback module.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲

   jax-tqdm github.com

   Add a tqdm progress bar to JAX scans and loops.

   tool link ↗
   
2. ▲

   Progress github.com

   Add beautiful progress bars to your loops.

   tool link ↗
   
3. ▲

   Writing an MCMC sampler in JAX by Jeremie Coullon jeremiecoullon.com

   Tutorial on the different ways to write an MCMC sampler in JAX along with speed benchmarks.

   site topic books & writing link ↗
   site
4. ▲

   progress github.com

   Flexible ascii progress bar.

   tool link ↗
   tool
5. ▲

   BlackJAX github.com

   Library of samplers for JAX.

   tool link ↗
   
6. ▲

   CircleProgress github.com

   Displays progress using a radial progressbar.

   tool link ↗
   

JAX › Tutorials and Blog Posts: “Tutorial on how to add a progress bar to compiled loops in JAX using the host_callback module.”