chore(deps): update dependency jax to >=0.4.35, <=0.4.35

[renovate]

This PR contains the following updates:

Package	Change	Age	Adoption	Passing	Confidence
jax	>=0.4.16, <=0.4.34 -> >=0.4.35, <=0.4.35

---

Release Notes

jax-ml/jax (jax)

v0.4.35

Compare Source

• Breaking Changes

  • {func}jax.numpy.isscalar now returns True for any array-like object with
    zero dimensions. Previously it only returned True for zero-dimensional
    array-like objects with a weak dtype.
  • jax.experimental.host_callback has been deprecated since March 2024, with
    JAX version 0.4.26. Now we removed it.
    See {jax-issue}#20385 for a discussion of alternatives.

• Changes:

  • jax.lax.FftType was introduced as a public name for the enum of FFT
    operations. The semi-public API jax.lib.xla_client.FftType has been
    deprecated.
  • TPU: JAX now installs TPU support from the libtpu package rather than
    libtpu-nightly. For the next few releases JAX will pin an empty version of
    libtpu-nightly as well as libtpu to ease the transition; that dependency
    will be removed in Q1 2025.

• Deprecations:

  • The semi-public API jax.lib.xla_client.PaddingType has been deprecated.
    No JAX APIs consume this type, so there is no replacement.
  • The default behavior of {func}jax.pure_callback and
    {func}jax.extend.ffi.ffi_call under vmap has been deprecated and so has
    the vectorized parameter to those functions. The vmap_method parameter
    should be used instead for better defined behavior. See the discussion in
    {jax-issue}#23881 for more details.
  • The semi-public API jax.lib.xla_client.register_custom_call_target has
    been deprecated. Use the JAX FFI instead.
  • The semi-public APIs jax.lib.xla_client.dtype_to_etype,
    jax.lib.xla_client.ops,
    jax.lib.xla_client.shape_from_pyval, jax.lib.xla_client.PrimitiveType,
    jax.lib.xla_client.Shape, jax.lib.xla_client.XlaBuilder, and
    jax.lib.xla_client.XlaComputation have been deprecated. Use StableHLO
    instead.

---

Configuration

📅 Schedule: Branch creation - At any time (no schedule defined), Automerge - At any time (no schedule defined).

🚦 Automerge: Disabled by config. Please merge this manually once you are satisfied.

♻ 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.

[github-actions]

Stale pull request message. Please comment to remove stale tag. Otherwise this pr will be closed soon.

[renovate]

Renovate Ignore Notification

Because you closed this PR without merging, Renovate will ignore this update (>=0.4.35, <=0.4.35). 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.