ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Make a progress_bar_fn that uses tqdm.

description str to display next to the progress bar, default is "".
leave Boolean whether to leave the progress bar up after finished.

tqdm_progress_bar_fn A function that takes an integer num_steps and returns a tqdm progress bar iterator.