change r2.5 to use jax stable version 0.4.33 (#8062)

pytorch · Oct 3, 2024 · c074e31 · c074e31
1 parent 31a0a92
commit c074e31
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -64,10 +64,10 @@
 
 base_dir = os.path.dirname(os.path.abspath(__file__))
 
-_date = '20240913'
+_date = '20240916'
 _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.dev{_date}'
+_jax_version = f'0.4.33'
 
 
 def _get_build_mode():