Bump patch version and update dependencies (jeremiecoullon#23)

Bump patch version (jeremiecoullon#21)

Update JAX dependency constraint, and bump minor version number (jere…

…miecoullon#18)

* Update JAX dependency constraint, and bump minor version number

* Update link in README to callback docs

Replace message arg with kwargs for tqdm (jeremiecoullon#13)

* replace message arg with tqdm kwargs

* added additional tqdm arguments example to readme

* lint with black

Add print-rate as optional argument (jeremiecoullon#11)

* Add print-rate as optional argument

* Update dependencies

* Update isort

* Update README, extend tests add checks on print-rate

* README tweaks from PR comments

* Fix typo

* Bump minor version number ahead of new release

Update README with installation instructions (jeremiecoullon#8)