deps: Update dependency jax to >=0.9.1

[renovate]

This PR contains the following updates:

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

---

Release Notes

jax-ml/jax (jax)

v0.9.1

Compare Source

• Changes:

  • JAX tracers that are not of Array type (e.g., of Ref type) will no
    longer report themselves to be instances of Array.
  • Using jax.shard_map in Explicit mode will raise an error
    if the PartitionSpec of input does not match the PartitionSpec specified in
    in_specs. In other words, it will act like an assert instead of an
    implicit reshard.
    in_specs is an optional argument so you can omit specifying it
    and shard_map will infer the PartitionSpec from the argument. If you
    want to reshard your inputs, you can use jax.reshard on the arguments and
    then pass those args to shard_map.

• New features:

  • Added a debug config jax_compilation_cache_check_contents. If set, we miss
    when get() is called on a value that has not been put() by the current
    process, even if the value is actually in the disk cache. When a value is
    put(), we verify that its contents match.

---

Configuration

📅 Schedule: (UTC)

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

🚦 Automerge: Enabled.

♻ Rebasing: Whenever PR is behind base branch, 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.

[renovate]

⚠️ Artifact update problem

Renovate failed to update an artifact related to this branch. You probably do not want to merge this PR as-is.

♻ Renovate will retry this branch, including artifacts, only when one of the following happens:

• any of the package files in this branch needs updating, or
• the branch becomes conflicted, or
• you click the rebase/retry checkbox if found above, or
• you rename this PR's title to start with "rebase!" to trigger it manually

The artifact failure details are included below:

File name: pixi.lock

ExecError: Command failed: pixi lock --no-progress --color=never --quiet
Error:   × failed to solve requirements of environment 'tests-cuda-py311' for
  │ platform 'osx-64'
  ├─▶   × failed to solve the environment
  │   
  ╰─▶ Cannot solve the request because of: jax >=0.9.1 cannot be installed
      because there are no viable options:
      └─ jax 0.9.1 would require
         └─ jaxlib >=0.9.1,<=0.9.1, for which no candidates were found.
      

[lucascolley]

this situation should be resolved once conda-forge/jax-feedstock#189 is in

[renovate]

Renovate Ignore Notification

Because you closed this PR without merging, Renovate will ignore this update (>=0.9.1). You will get a PR once a newer version is released. To ignore this dependency forever, add it to the ignoreDeps array of your Renovate config.

If you accidentally closed this PR, or if you changed your mind: rename this PR to get a fresh replacement PR.