deps: Update dependency jax to >=0.9.2

[renovate]

This PR contains the following updates:

Package	Type	Update	Change
jax	feature-backends	patch	>=0.9.1 → >=0.9.2

---

Release Notes

jax-ml/jax (jax)

v0.9.2

Compare Source

• Changes:
  • The semi-private type jax._src.literals.TypedNdArray is now a subclass of
    np.ndarray, rather than a duck type of it.
  • {func}jax.numpy.arange with step specified no longer generates the array
    on host. The benefit is more efficient code, though this can lead to less
    precise outputs for narrow-width floats (e.g. bfloat16). To recover the
    previous behavior in this case, use jnp.array(np.arange(...)).

---

Configuration

📅 Schedule: (UTC)

• Branch creation
  • At any time (no schedule defined)
• Automerge
  • At any time (no schedule defined)

🚦 Automerge: Enabled.

♻ Rebasing: Whenever PR becomes conflicted, or you tick the rebase/retry checkbox.

🔕 Ignore: Close this PR and you won't be reminded about this update again.

---

• If you want to rebase/retry this PR, check this box

---

This PR was generated by Mend Renovate. View the repository job log.