From 69d436433b84c66d2bd766e352e65f7bb9653ec8 Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Tue, 24 Sep 2024 01:42:46 -0700
Subject: [PATCH 1/5] Update jax to stable 0.4.33 version

---
 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():

From 48bdcaed68732c54f164a8e1640cb4aa20fed418 Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Tue, 24 Sep 2024 10:11:38 -0700
Subject: [PATCH 2/5] revert change of jax version in cherry-pick PR

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

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

From c21a643c2798fa69cf19b46c9b58c2c55d03d143 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 3/5] 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():

From a141ecf4522eaf734c911b02c5d6eb914c7e25fc Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Fri, 27 Sep 2024 10:23:27 -0700
Subject: [PATCH 4/5] Update setup.py

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

diff --git a/setup.py b/setup.py
index 43fe7e81ed4..2d85cb8d7b4 100644
--- a/setup.py
+++ b/setup.py
@@ -70,6 +70,7 @@
 _jax_version = f'0.4.33'
 
 
+
 def _get_build_mode():
   for i in range(1, len(sys.argv)):
     if not sys.argv[i].startswith('-'):

From 3a27d689f99fb2a7bf6b59ce67b4430ad453997e Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Fri, 27 Sep 2024 10:23:37 -0700
Subject: [PATCH 5/5] Update setup.py

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

diff --git a/setup.py b/setup.py
index 2d85cb8d7b4..43fe7e81ed4 100644
--- a/setup.py
+++ b/setup.py
@@ -70,7 +70,6 @@
 _jax_version = f'0.4.33'
 
 
-
 def _get_build_mode():
   for i in range(1, len(sys.argv)):
     if not sys.argv[i].startswith('-'):