Speed up Schrodinger Follmer test (#741)

* Plotting BlackJAX with BlackJAX

* Plotting BlackJAX with BlackJAX

* Update blackjax/mcmc/metrics.py

Co-authored-by: Junpeng Lao <junpenglao@gmail.com>

* Update blackjax/mcmc/metrics.py

Co-authored-by: Junpeng Lao <junpenglao@gmail.com>

* Merged comments from Junpeng

* Speed up Follmer

---------

Co-authored-by: Junpeng Lao <junpenglao@gmail.com>

remove labels (#716)

Fix doc mistakes (#701)

* Fix equation formatting

* Clarify JAX gradient error

* Fix punctuation + capitalization

* Fix grammar

Should not begin sentence with "i.e." in English.

* Fix math formatting error

* Fix typo

Change parallel _ensample_ chain adaptation to parallel _ensemble_ chain adaptation.

* Add SVGD citation to appear in doc

Currently the SVGD paper is only cited in the `kernel` function, which is defined _within_ the `build_kernel` function. Because of this nested function format, the SVGD paper is _not_ cited in the documentation.

To fix this, I added a citation to the SVGD paper in the `as_top_level_api` docstring.

* Fix grammar + clarify doc

* Fix typo

---------

Co-authored-by: Junpeng Lao <junpenglao@gmail.com>

Update jax version requirements (#666)

Fix #665

 Removal of Algorithm classes. (#657)

* more

* removing export

* removal of classes, tests passing

* linter

* fix on test

* linter

* removing parametrization on test

* code review updates

* exporting as_top_level_api in dynamic_hmc

* linter

* code review update: replace imports

Update README.md (#638)

* Update README.md

Update citation.

* Update README.md

PREVENT 1D DISTRIBUTION (#633)

* PREVENT 1D DISTRIBUTION

* PREVENT 1D DISTRIBUTION

Update dependencies and use new jax random key (#569)

* Update dependency

* chex version typo

* fix typing

* fix typing 2

* Revert "fix typing 2"

This reverts commit 154e6b3.

* jax.random.PRNGKey(...) -> jax.random.key(...)

Fix documentation typo for kernel and state input

Add SGHMC