blackjax.progress_bar

Contents

`blackjax.progress_bar`#

Progress bar decorators for use with step functions. Adapted from Jeremie Coullon’s blog post [Cou].

Module Contents#

Functions#

progress_bar_scan(num_samples[, print_rate])

Progress bar for a JAX scan

progress_bar_scan(num_samples, print_rate=None)[source]#: Progress bar for a JAX scan