GitHub repository: mdmould/jax-tqdm
Description: Add a tqdm progress bar to your JAX scans and loops.