From a4a5b0802308d68a618184093db2399d4fd8aacc Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Tue, 24 Sep 2024 10:22:15 -0700
Subject: [PATCH] Update setup.py

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index f8eb3f61dea..43fe7e81ed4 100644
--- 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.dev{_date}'
+_jax_version = f'0.4.33'
 
 
 def _get_build_mode():