Current requirements allows for jax >=0.4.25, but new jnp.clip argument requires jax 0.4.27

[juehang]

blackjax/requirements-doc.txt

Line 7 in 83bc3a0

jax>=0.4.25

This line is causing the issue. Lines where jnp.clip is used were recently updated due to #663 but currently the requirements allow the latest version of blackjax to be installed with versions of jax that do not recognise the new keyword argument. I encounter this with jax 0.4.26, provided with the Nvidia NGC jax docker image.