potential jax 0.4.35 release issue?

[haohuanw]

Description

i am checking jax 0.4.35's pip wheel metadata and found that cuda12-pip variant depends on jaxlib==0.4.34 is this expected?

Metadata-Version: 2.1
Name: jax
Version: 0.4.35
Summary: Differentiate, compile, and transform Numpy code.
Home-page: https://github.com/jax-ml/jax
Author: JAX team
Author-email: [email protected]
License: Apache-2.0
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Requires-Python: >=3.10
Description-Content-Type: text/markdown
License-File: LICENSE
License-File: AUTHORS
Requires-Dist: jaxlib<=0.4.35,>=0.4.34
Requires-Dist: ml-dtypes>=0.4.0
Requires-Dist: numpy>=1.24
Requires-Dist: opt-einsum
Requires-Dist: scipy>=1.10
Requires-Dist: numpy>=1.26.0; python_version >= "3.12"
Requires-Dist: scipy>=1.11.1; python_version >= "3.12"
Provides-Extra: ci
Requires-Dist: jaxlib==0.4.34; extra == "ci"
Provides-Extra: cpu
Provides-Extra: cuda
Requires-Dist: jaxlib==0.4.34; extra == "cuda"
Requires-Dist: jax-cuda12-plugin[with_cuda]<=0.4.35,>=0.4.34; extra == "cuda"
Provides-Extra: cuda12
Requires-Dist: jaxlib==0.4.34; extra == "cuda12"
Requires-Dist: jax-cuda12-plugin[with_cuda]<=0.4.35,>=0.4.34; extra == "cuda12"
Provides-Extra: cuda12_local
Requires-Dist: jaxlib==0.4.34; extra == "cuda12-local"
Requires-Dist: jax-cuda12-plugin==0.4.34; extra == "cuda12-local"
Provides-Extra: cuda12_pip
Requires-Dist: jaxlib==0.4.34; extra == "cuda12-pip"   **<= section in question**
Requires-Dist: jax-cuda12-plugin[with_cuda]<=0.4.35,>=0.4.34; extra == "cuda12-pip"
Provides-Extra: k8s
Requires-Dist: kubernetes; extra == "k8s"
Provides-Extra: minimum-jaxlib
Requires-Dist: jaxlib==0.4.34; extra == "minimum-jaxlib"
Provides-Extra: tpu
Requires-Dist: jaxlib<=0.4.35,>=0.4.34; extra == "tpu"
Requires-Dist: libtpu-nightly==0.1.dev20241010+nightly.cleanup; extra == "tpu"
Requires-Dist: libtpu==0.0.2; extra == "tpu"
Requires-Dist: requests; extra == "tpu"

System info (python version, jaxlib version, accelerator, etc.)

N/A

[mattjj]

Thanks for raising this! Sounds like it's a known issue and we're going to fix it in the next release.