Please provide updated example to use HMC in Blackjax

[grahnder]

Can someone please provide a demo notebook to explain how the api can be used to perform HMC on a probabilistic model. All the example notebooks have outdated.

[junpenglao]

We are in the process of updating the documentations, created an issue for tracking: #551

[junpenglao]

Actually, the notebooks are tested against main:

If you install from main or nightly release the example should be working.