diff --git a/setup.py b/setup.py index 43fe7e81ed4..f8eb3f61dea 100644 --- a/setup.py +++ b/setup.py @@ -67,7 +67,7 @@ _date = '20240913' _libtpu_version = f'0.1.dev{_date}' _libtpu_storage_path = f'https://storage.googleapis.com/libtpu-nightly-releases/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}+nightly-py3-none-any.whl' -_jax_version = f'0.4.33' +_jax_version = f'0.4.33.dev{_date}' def _get_build_mode():