Bump JAX version in setup.py (#6740)

pytorch · Mar 13, 2024 · 977e7ef · 977e7ef
1 parent 1bbe333
commit 977e7ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -67,7 +67,7 @@
 _date = '20240305'
 _libtpu_version = f'0.1.dev{_date}'
 _libtpu_storage_path = f'https://storage.googleapis.com/cloud-tpu-tpuvm-artifacts/wheels/libtpu-nightly/libtpu_nightly-{_libtpu_version}-py3-none-any.whl'
-_jax_version = f'0.4.25.dev{_date}'
+_jax_version = f'0.4.26.dev{_date}'
 
 
 def _get_build_mode():