From 4443be32b17e40ce0f36df85f92a9ef17d536de1 Mon Sep 17 00:00:00 2001 From: Will Cromar Date: Wed, 13 Mar 2024 12:48:30 -0700 Subject: [PATCH] Bump JAX version in `setup.py` --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 4d26e1a24606..249dbb307c15 100644 --- 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():