revert change of jax version in cherry-pick PR

pytorch · Sep 27, 2024 · 48bdcae · 48bdcae
1 parent 69d4364
commit 48bdcae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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():