Fix documentation typo for kernel and state input

Add SGHMC

Use Flax in the Hierarchical BNN example

Fix the workflow that publishes the docs

Bump to 0.9.2

Bump version 0.9.1

Bump Blackjax to 0.9.0

Bump Blackjax version to 0.8.3

Bump version to 0.8.2

Bump version to 0.8.1