Select Tqdm Submodule

Adds the ability to select the the tqdm submodule, for example manually selecting 'std' or 'notebook' for progress bar creation (thanks @mdmould)

Bug Fixes

• Fixes #19 for JAX version > 0.4.27 (thanks @mdmould)
• Fix async update bug (mentioned in #12) by ensuring tqdm updates are ordered (thanks also @mdmould)

JAX Update

• Use jax.debug.callback instead of deprecated jax.experimental.host_callback (thanks @BirkhoffG )
• Update min JAX version to 0.4.12

Progress Bar Options

Adds the ability to pass tqdm keyword options as arguments (thanks to @mdmould )

v0.1.1

Add the ability to manually set print-rate

Initial Release

Implements tqdm progress bars for JAX scans and for loops.