Skip to content

Progress bar for fori_loop #13126

Answered by sharadmv
rezaaskary asked this question in Ideas
Discussion options

You must be logged in to vote

I created an example of you how can use jax.debug.callback to create a progress bar system for JAX:

https://gist.github.com/sharadmv/be24b3107cf9b8bf027ea8e2f177882e

Does this work?

Replies: 1 comment 1 reply

Comment options

You must be logged in to vote
1 reply
@rezaaskary
Comment options

Answer selected by rezaaskary
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Ideas
Labels
None yet
2 participants