From 1594d2f30fdbfebf693aba4a2b264e4a3e52acc6 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Wed, 11 Sep 2024 11:58:03 -0400
Subject: [PATCH] Prepare for v0.4.32 release.

---
 .github/workflows/wheel_win_x64.yml | 2 +-
 jax/version.py                      | 2 +-
 setup.py                            | 4 ++--
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/wheel_win_x64.yml b/.github/workflows/wheel_win_x64.yml
index 61912ed8978e..03b5de37b85b 100644
--- a/.github/workflows/wheel_win_x64.yml
+++ b/.github/workflows/wheel_win_x64.yml
@@ -58,7 +58,7 @@ jobs:
           JAX_SKIP_SLOW_TESTS: true
           PY_COLORS: 1
         run: |
+          python -m pip install --find-links ${{ github.workspace }}\dist jaxlib
           python -m pip install -e ${{ github.workspace }}
-          python -m pip install --no-index --find-links ${{ github.workspace }}\dist jaxlib
           echo "JAX_ENABLE_CHECKS=$JAX_ENABLE_CHECKS"
           pytest -n auto --tb=short tests examples
diff --git a/jax/version.py b/jax/version.py
index cc690e02cb46..f2c34d275b01 100644
--- a/jax/version.py
+++ b/jax/version.py
@@ -133,7 +133,7 @@ def make_release_tree(self, base_dir, files):
 
 
 __version__ = _get_version_string()
-_minimum_jaxlib_version = "0.4.31"
+_minimum_jaxlib_version = "0.4.32"
 
 def _version_as_tuple(version_str):
   return tuple(int(i) for i in version_str.split(".") if i.isdigit())
diff --git a/setup.py b/setup.py
index 08ce8dbcb4ed..027e5aefbc2f 100644
--- a/setup.py
+++ b/setup.py
@@ -19,10 +19,10 @@
 
 project_name = 'jax'
 
-_current_jaxlib_version = '0.4.31'
+_current_jaxlib_version = '0.4.32'
 # The following should be updated after each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.4.31'
-_libtpu_version = '0.1.dev20240729'
+_libtpu_version = '0.1.dev20240911'
 
 def load_version_module(pkg_path):
   spec = importlib.util.spec_from_file_location(