Skip to content

deps: Update jax to >=0.7.0,!=0.6.2 #386

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Closed
wants to merge 1 commit into from
Closed

deps: Update jax to >=0.7.0,!=0.6.2 #386

wants to merge 1 commit into from

Conversation

@renovate
Copy link
Contributor

@renovate renovate bot commented Aug 8, 2025 •
edited
Loading

This PR contains the following updates:

Package Type Update Change
jax feature-backends minor >=0.6.0,!=0.6.2 -> >=0.7.0,!=0.6.2
jaxlib feature-cuda-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 is behind base branch, or you tick the rebase/retry checkbox.

🔕 Ignore: Close this PR and you won't be reminded about these updates 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 renovate bot added the dependencies Pull requests that update a dependency file label Aug 8, 2025
@renovate
Copy link
Contributor Author

renovate bot commented Aug 8, 2025 •
edited
Loading

⚠️ 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.0/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 'backends' for platform 'osx-
  │ 64'
  ├─▶ failed to solve the environment
  ╰─▶ Cannot solve the request because of: The following packages are
      incompatible
      ├─ jax >=0.7.0,!=0.6.2 can be installed with any of the following
      options:
      │  └─ jax 0.7.0 would require
      │     └─ jaxlib >=0.7.0,<=0.7.0, which can be installed with any of the
      following options:
      │        └─ jaxlib 0.7.0 would require
      │           └─ python_abi 3.11.* *_cp311, which can be installed with
      any of the following options:
      │              └─ python_abi 3.11
      └─ python ~=3.10.0 cannot be installed because there are no viable
      options:
         └─ python 3.10.0 | 3.10.0 | 3.10.0 | 3.10.0 | 3.10.0 | 3.10.0 |
      3.10.1 | 3.10.1 | 3.10.1 | 3.10.1 | 3.10.1 | 3.10.1 | 3.10.2 | 3.10.2
      | 3.10.2 | 3.10.2 | 3.10.2 | 3.10.2 | 3.10.2 | 3.10.2 | 3.10.2 | 3.10.2
      | 3.10.4 | 3.10.4 | 3.10.5 | 3.10.5 | 3.10.6 | 3.10.6 | 3.10.7 | 3.10.7
      | 3.10.8 | 3.10.8 | 3.10.9 | 3.10.10 | 3.10.11 | 3.10.12 | 3.10.13 |
      3.10.13 | 3.10.14 | 3.10.15 | 3.10.15 | 3.10.15 | 3.10.16 | 3.10.16 |
      3.10.16 | 3.10.17 | 3.10.18 | 3.10.18 would constrain
            └─ python_abi 3.10.* *_cp310, which conflicts with any installable
      versions previously reported

@renovate renovate bot force-pushed the renovate/jax branch 4 times, most recently from 147e6ac to 92b6ec1 Compare August 10, 2025 08:21
@renovate renovate bot changed the title deps: Update dependency jaxlib to >=0.6.2 deps: Update dependency jaxlib to >=0.7.0 Aug 10, 2025
@renovate renovate bot changed the title deps: Update dependency jaxlib to >=0.7.0 deps: Update jax to >=0.7.0,!=0.6.2 Aug 13, 2025
@renovate renovate bot force-pushed the renovate/jax branch 2 times, most recently from 9fecbc7 to 8e2e52d Compare August 15, 2025 04:21
@lucascolley
Copy link
Member

lucascolley commented Aug 15, 2025

blocked by dropping Python 3.10

@renovate
Copy link
Contributor Author

renovate bot commented Aug 16, 2025

Renovate Ignore Notification

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

@renovate renovate bot deleted the renovate/jax branch August 16, 2025 12:34
@lucascolley lucascolley mentioned this pull request Aug 16, 2025
Open
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

dependencies Pull requests that update a dependency file

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

2 participants

@lucascolley