deps: Update dependency jax to >=0.7.0 - autoclosed

[renovate]

This PR contains the following updates:

Package	Type	Update	Change
jax	feature-backends	minor	>=0.6.0 -> >=0.7.0

---

Release Notes

jax-ml/jax (jax)

v0.7.0

Compare Source

• New features:

  • Added jax.P which is an alias for jax.sharding.PartitionSpec.
  • Added {func}jax.tree.reduce_associative.
  • The {attr}jax.numpy.ndarray.at indexing methods now support a wrap_negative_indices
    argument, which defaults to True to match the current behavior ({jax-issue}#29434).

• Breaking changes:

  • JAX is migrating from GSPMD to Shardy by default. See the
    migration guide
    for more information.
  • JAX autodiff is switching to using direct linearization by default (instead of
    implementing linearization via JVP and partial eval).
    See migration guide
    for more information.
  • jax.stages.OutInfo has been replaced with jax.ShapeDtypeStruct.
  • {func}jax.jit now requires fun to be passed by position, and additional
    arguments to be passed by keyword. Doing otherwise will result in an error
    starting in v0.7.x. This raised a DeprecationWarning in v0.6.x.
  • The minimum Python version is now 3.11. 3.11 will remain the minimum
    supported version until July 2026.
  • Layout API renames:
    • Layout, .layout, .input_layouts and .output_layouts have been
      renamed to Format, .format, .input_formats and .output_formats
    • DeviceLocalLayout, .device_local_layout have been renamed to Layout
      and .layout
  • jax.experimental.shard module has been deleted and all the APIs have been
    moved to the jax.sharding endpoint. So use jax.sharding.reshard,
    jax.sharding.auto_axes and jax.sharding.explicit_axes instead of their
    experimental endpoints.
  • lax.infeed and lax.outfeed were removed, after being deprecated in
    JAX 0.6. The transfer_to_infeed and transfer_from_outfeed methods were
    also removed the Device objects.
  • The jax.extend.core.primitives.pjit_p primitive has been renamed to
    jit_p, and its name attribute has changed from "pjit" to "jit".
    This affects the string representations of jaxprs. The same primitive is no
    longer exported from the jax.experimental.pjit module.
  • The (undocumented) function jax.extend.backend.add_clear_backends_callback
    has been removed. Users should use jax.extend.backend.register_backend_cache
    instead.
  • out_sharding arg added to x.at[y].set and x.at[y].add. Previous
    behavior propagating operand sharding removed. Please use
    x.at[y].set/add(z, out_sharding=jax.typeof(x).sharding) to retain previous
    behavior if scatter op requires collectives.

• Deprecations:

  • {obj}jax.dlpack.SUPPORTED_DTYPES is deprecated; please use the new
    {func}jax.dlpack.is_supported_dtype function.
  • {func}jax.scipy.special.sph_harm has been deprecated following a similar
    deprecation in SciPy; use {func}jax.scipy.special.sph_harm_y instead.
  • From {mod}jax.interpreters.xla, the previously deprecated symbols
    abstractify and pytype_aval_mappings have been removed.
  • {func}jax.interpreters.xla.canonicalize_dtype is deprecated. For
    canonicalizing dtypes, prefer {func}jax.dtypes.canonicalize_dtype.
    For checking whether an object is a valid jax input, prefer
    {func}jax.core.valid_jaxtype.
  • From {mod}jax.core, the previously deprecated symbols AxisName,
    ConcretizationTypeError, axis_frame, call_p, closed_call_p,
    get_type, trace_state_clean, typematch, and typecheck have been
    removed.
  • From {mod}jax.lib.xla_client, the previously deprecated symbols
    DeviceAssignment, get_topology_for_devices, and mlir_api_version
    have been removed.
  • jax.extend.ffi was removed after being deprecated in v0.5.0.
    Use {mod}jax.ffi instead.
  • {func}jax.lib.xla_bridge.get_compile_options is deprecated, and replaced by
    {func}jax.extend.backend.get_compile_options.

v0.6.2

Compare Source

• New features:

  • Added {func}jax.tree.broadcast which implements a pytree prefix broadcasting helper.

• Changes

  • The minimum NumPy version is 1.26 and the minimum SciPy version is 1.12.

---

Configuration

📅 Schedule: 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.

[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

thread 'tokio-runtime-worker' panicked at /home/runner/.cargo/registry/src/index.crates.io-1949cf8c6b5b557f/tokio-1.47.1/src/runtime/time/entry.rs:602:9:
A Tokio 1.x context was found, but it is being shutdown.
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace
Error:   × failed to solve requirements of environment 'tests-cuda-py310' for
  │ platform 'osx-arm64'
  ├─▶   × failed to solve the environment
  │   
  ╰─▶ Cannot solve the request because of: The following packages are
      incompatible
      ├─ python ~=3.10.0 cannot be installed because there are no viable
      options:
      │  ├─ python 3.10.18, which conflicts with the versions reported above.
      │  ├─ python 3.10.18, which conflicts with the versions reported above.
      │  ├─ python 3.10.17, which conflicts with the versions reported above.
      │  ├─ python 3.10.16, which conflicts with the versions reported above.
      │  ├─ python 3.10.16, which conflicts with the versions reported above.
      │  ├─ python 3.10.16, which conflicts with the versions reported above.
      │  ├─ python 3.10.15, which conflicts with the versions reported above.
      │  ├─ python 3.10.15, which conflicts with the versions reported above.
      │  ├─ python 3.10.15, which conflicts with the versions reported above.
      │  ├─ python 3.10.14, which conflicts with the versions reported above.
      │  ├─ python 3.10.13, which conflicts with the versions reported above.
      │  ├─ python 3.10.13, which conflicts with the versions reported above.
      │  ├─ python 3.10.12, which conflicts with the versions reported above.
      │  ├─ python 3.10.11, which conflicts with the versions reported above.
      │  ├─ python 3.10.10, which conflicts with the versions reported above.
      │  ├─ python 3.10.9, which conflicts with the versions reported above.
      │  ├─ python 3.10.8, which conflicts with the versions reported above.
      │  ├─ python 3.10.8, which conflicts with the versions reported above.
      │  ├─ python 3.10.7, which conflicts with the versions reported above.
      │  ├─ python 3.10.7, which conflicts with the versions reported above.
      │  ├─ python 3.10.6, which conflicts with the versions reported above.
      │  ├─ python 3.10.6, which conflicts with the versions reported above.
      │  ├─ python 3.10.5, which conflicts with the versions reported above.
      │  ├─ python 3.10.5, which conflicts with the versions reported above.
      │  ├─ python 3.10.4, which conflicts with the versions reported above.
      │  ├─ python 3.10.4, which conflicts with the versions reported above.
      │  ├─ python 3.10.2, which conflicts with the versions reported above.
      │  ├─ python 3.10.2, which conflicts with the versions reported above.
      │  ├─ python 3.10.2, which conflicts with the versions reported above.
      │  ├─ python 3.10.2, which conflicts with the versions reported above.
      │  ├─ python 3.10.2, which conflicts with the versions reported above.
      │  ├─ python 3.10.2, which conflicts with the versions reported above.
      │  ├─ python 3.10.2, which conflicts with the versions reported above.
      │  ├─ python 3.10.2, which conflicts with the versions reported above.
      │  ├─ python 3.10.2, which conflicts with the versions reported above.
      │  ├─ python 3.10.2, which conflicts with the versions reported above.
      │  ├─ python 3.10.1, which conflicts with the versions reported above.
      │  ├─ python 3.10.1, which conflicts with the versions reported above.
      │  ├─ python 3.10.1, which conflicts with the versions reported above.
      │  ├─ python 3.10.1, which conflicts with the versions reported above.
      │  ├─ python 3.10.1, which conflicts with the versions reported above.
      │  ├─ python 3.10.1, which conflicts with the versions reported above.
      │  ├─ python 3.10.0, which conflicts with the versions reported above.
      │  ├─ python 3.10.0, which conflicts with the versions reported above.
      │  ├─ python 3.10.0, which conflicts with the versions reported above.
      │  ├─ python 3.10.0, which conflicts with the versions reported above.
      │  ├─ python 3.10.0, which conflicts with the versions reported above.
      │  └─ python 3.10.0, which conflicts with the versions reported above.
      └─ jax >=0.7.0 cannot be installed because there are no viable options:
         └─ jax 0.7.0 would require
            └─ python >=3.11, which cannot be installed because there are no
      viable options:
               ├─ python 3.13.7, which conflicts with the versions reported
      above.
               ├─ python 3.13.5, which conflicts with the versions reported
      above.
               ├─ python 3.13.5, which conflicts with the versions reported
      above.
               ├─ python 3.13.5, which conflicts with the versions reported
      above.
               ├─ python 3.13.3, which conflicts with the versions reported
      above.
               ├─ python 3.13.3, which conflicts with the versions reported
      above.
               ├─ python 3.13.2, which conflicts with the versions reported
      above.
               ├─ python 3.13.2, which conflicts with the versions reported
      above.
               ├─ python 3.13.1, which conflicts with the versions reported
      above.
               ├─ python 3.13.1, which conflicts with the versions reported
      above.
               ├─ python 3.13.1, which conflicts with the versions reported
      above.
               ├─ python 3.13.1, which conflicts with the versions reported
      above.
               ├─ python 3.13.1, which conflicts with the versions reported
      above.
               ├─ python 3.13.1, which conflicts with the versions reported
      above.
               ├─ python 3.13.0, which conflicts with the versions reported
      above.
               ├─ python 3.13.0, which conflicts with the versions reported
      above.
               ├─ python 3.13.0, which conflicts with the versions reported
      above.
               ├─ python 3.12.11, which conflicts with the versions reported
      above.
               ├─ python 3.12.10, which conflicts with the versions reported
      above.
               ├─ python 3.12.9, which conflicts with the versions reported
      above.
               ├─ python 3.12.9, which conflicts with the versions reported
      above.
               ├─ python 3.12.8, which conflicts with the versions reported
      above.
               ├─ python 3.12.8, which conflicts with the versions reported
      above.
               ├─ python 3.12.7, which conflicts with the versions reported
      above.
               ├─ python 3.12.6, which conflicts with the versions reported
      above.
               ├─ python 3.12.6, which conflicts with the versions reported
      above.
               ├─ python 3.12.6, which conflicts with the versions reported
      above.
               ├─ python 3.12.5, which conflicts with the versions reported
      above.
               ├─ python 3.12.4, which conflicts with the versions reported
      above.
               ├─ python 3.12.3, which conflicts with the versions reported
      above.
               ├─ python 3.12.2, which conflicts with the versions reported
      above.
               ├─ python 3.12.1, which conflicts with the versions reported
      above.
               ├─ python 3.12.1, which conflicts with the versions reported
      above.
               ├─ python 3.12.0, which conflicts with the versions reported
      above.
               ├─ python 3.11.13, which conflicts with the versions reported
      above.
               ├─ python 3.11.12, which conflicts with the versions reported
      above.
               ├─ python 3.11.11, which conflicts with the versions reported
      above.
               ├─ python 3.11.11, which conflicts with the versions reported
      above.
               ├─ python 3.11.11, which conflicts with the versions reported
      above.
               ├─ python 3.11.10, which conflicts with the versions reported
      above.
               ├─ python 3.11.10, which conflicts with the versions reported
      above.
               ├─ python 3.11.10, which conflicts with the versions reported
      above.
               ├─ python 3.11.10, which conflicts with the versions reported
      above.
               ├─ python 3.11.9, which conflicts with the versions reported
      above.
               ├─ python 3.11.8, which conflicts with the versions reported
      above.
               ├─ python 3.11.7, which conflicts with the versions reported
      above.
               ├─ python 3.11.7, which conflicts with the versions reported
      above.
               ├─ python 3.11.6, which conflicts with the versions reported
      above.
               ├─ python 3.11.5, which conflicts with the versions reported
      above.
               ├─ python 3.11.4, which conflicts with the versions reported
      above.
               ├─ python 3.11.3, which conflicts with the versions reported
      above.
               ├─ python 3.11.2, which conflicts with the versions reported
      above.
               ├─ python 3.11.1, which conflicts with the versions reported
      above.
               ├─ python 3.11.0, which conflicts with the versions reported
      above.
               ├─ python 3.11.0, which conflicts with the versions reported
      above.
               ├─ python 3.11.0, which conflicts with the versions reported
      above.
               ├─ python 3.11.0, which conflicts with the versions reported
      above.
               ├─ python 3.14.0rc1 | 3.14.0rc1 | 3.14.0rc2 | 3.14.0rc2 would
      require
               │  └─ _python_rc *, for which no candidates were found.
               ├─ python 3.13.7, which conflicts with the versions reported
      above.
               ├─ python 3.13.5, which conflicts with the versions reported
      above.
               ├─ python 3.13.5, which conflicts with the versions reported
      above.
               ├─ python 3.13.5, which conflicts with the versions reported
      above.
               ├─ python 3.13.3, which conflicts with the versions reported
      above.
               ├─ python 3.13.3, which conflicts with the versions reported
      above.
               ├─ python 3.13.2, which conflicts with the versions reported
      above.
               ├─ python 3.13.2, which conflicts with the versions reported
      above.
               ├─ python 3.13.1, which conflicts with the versions reported
      above.
               ├─ python 3.13.1, which conflicts with the versions reported
      above.
               ├─ python 3.13.1, which conflicts with the versions reported
      above.
               ├─ python 3.13.1, which conflicts with the versions reported
      above.
               ├─ python 3.13.1, which conflicts with the versions reported
      above.
               ├─ python 3.13.1, which conflicts with the versions reported
      above.
               ├─ python 3.13.0, which conflicts with the versions reported
      above.
               ├─ python 3.13.0, which conflicts with the versions reported
      above.
               └─ python 3.13.0, which conflicts with the versions reported
      above.
      

[lucascolley]

we should add something akin to

array-api-extra/renovate.json

Lines 29 to 34 in db75e01

	"description": "Block PRs for updates blocked on dropping Python 3.10.",
	"matchManagers": ["pixi"],
	"matchUpdateTypes": ["major", "minor"],
	"matchPackageNames": ["numpy", "sphinx", "ipython", "sphinx-autodoc-typehints"],
	"enabled": false
	}, {

for Python 3.11 I think

[crusaderky]

Do not merge.
jax 0.7.0 and later is incompatible with Python 3.10.
jax.0.6.2 and later segfaults on CUDA. Note that CUDA is not in CI.