Merge pull request #17796 from hawkinsp:changelog

PiperOrigin-RevId: 568689288

CHANGELOG.md

@@ -19,9 +19,19 @@ Remember to align the itemized text with the first line of an item within a list
   * {func}`jax.scipy.stats.mode` now returns a 0 count if the mode is taken
     across a size-0 axis, matching the behavior of `scipy.stats.mode` in SciPy
     1.11.
+  * Removed the "No GPU/TPU" found warning. Instead warn if, on Linux, an
+    NVIDIA GPU or a Google TPU are found but not used and `--jax_platforms` was
+    not specified.
 
 # jaxlib 0.4.17
 
+* Changes:
+  * Python 3.12 wheels were added in this release.
+  * The CUDA 12 wheels now require CUDA 12.2.
+
+* Bug fixes:
+  * Fixed log spam from ABSL when the JAX CPU backend was initialized.
+
 ## jax 0.4.16 (Sept 18, 2023)
 
 * Changes

setup.py

@@ -123,14 +123,14 @@ def generate_proto(source):
 
         'cuda12_pip': [
           f"jaxlib=={_current_jaxlib_version}+cuda12.cudnn{_default_cuda12_cudnn_version}",
-          "nvidia-cublas-cu12",
-          "nvidia-cuda-cupti-cu12",
-          "nvidia-cuda-nvcc-cu12",
-          "nvidia-cuda-runtime-cu12",
+          "nvidia-cublas-cu12>=12.2.5.6",
+          "nvidia-cuda-cupti-cu12>=12.2.142",
+          "nvidia-cuda-nvcc-cu12>=12.2.140",
+          "nvidia-cuda-runtime-cu12>=12.2.140",
           "nvidia-cudnn-cu12>=8.9",
-          "nvidia-cufft-cu12",
-          "nvidia-cusolver-cu12",
-          "nvidia-cusparse-cu12",
+          "nvidia-cufft-cu12>=11.0.8.103",
+          "nvidia-cusolver-cu12>=11.5.2",
+          "nvidia-cusparse-cu12>=12.1.2.141",
         ],
 
         # Target that does not depend on the CUDA pip wheels, for those who want