From acf155feb589ed3d8ea387d43a10c78bf87374cd Mon Sep 17 00:00:00 2001
From: gkroiz <gersonkroiz@google.com>
Date: Sat, 17 Feb 2024 00:18:36 +0000
Subject: [PATCH] pin dependencies for 1.3.0 release

---
 paxml/pip_package/requirements.txt | 8 ++++----
 requirements.in                    | 8 ++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/paxml/pip_package/requirements.txt b/paxml/pip_package/requirements.txt
index abc0b96b5..18667ddd1 100644
--- a/paxml/pip_package/requirements.txt
+++ b/paxml/pip_package/requirements.txt
@@ -75,7 +75,7 @@ click==8.1.7
     #   nltk
     #   tensorflow-datasets
     #   tfds-nightly
-clu @ git+https://github.com/google/CommonLoopUtils
+clu==0.0.11
     # via
     #   -r paxml-requirements.in
     #   -r praxis-requirements.in
@@ -128,7 +128,7 @@ executing==2.0.1
     # via stack-data
 fastjsonschema==2.19.1
     # via nbformat
-fiddle @ git+https://github.com/google/fiddle
+fiddle==0.2.11
     # via -r praxis-requirements.in
 filelock==3.13.1
     # via
@@ -136,7 +136,7 @@ filelock==3.13.1
     #   transformers
 flatbuffers==1.12
     # via tensorflow
-flax @ git+https://github.com/google/flax
+flax==0.8.1
     # via
     #   -r paxml-requirements.in
     #   -r praxis-requirements.in
@@ -215,7 +215,7 @@ ipywidgets==8.1.2
     # via jupyter
 isoduration==20.11.0
     # via jsonschema
-jax @ git+https://github.com/google/jax
+jax==0.4.24
     # via
     #   -r paxml-requirements.in
     #   -r praxis-requirements.in
diff --git a/requirements.in b/requirements.in
index 1d3fb07a3..cd320940e 100644
--- a/requirements.in
+++ b/requirements.in
@@ -2,14 +2,14 @@
 # bash ./compile_requirements.sh
 
 absl-py
-clu @ git+https://github.com/google/CommonLoopUtils
+clu
 etils
-flax @ git+https://github.com/google/flax
+flax==0.8.1
 graphviz
-jax @ git+https://github.com/google/jax
+jax==0.4.24
 lingvo
 numpy
-orbax-checkpoint
+orbax-checkpoint==0.5.3
 praxis
 protobuf==3.19.6
 pyglove