https://github.com/jeremiecoullon/jax-tqdm
Python47
5 months ago
Add a tqdm progress bar to your JAX scans and loops.
MIT License