Backport Bump JAX version in setup.py (#6740) (#6741)

Co-authored-by: Will Cromar <wcromar@google.com>
pytorch · Mar 13, 2024 · a47f81d · a47f81d
1 parent aa6d277
commit a47f81d
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():