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(