prod: lower bound for jax to 0.6

[beckermr]

I think this will work fine and will help the package be more compatible.

[codspeed-hq]

Merging this PR will not alter performance

✅ 16 untouched benchmarks

---

_{Comparing try-lower-bound (c0f2694) with main (513a68c)}