From 6143a74d2c893afdb63cc67c236e56899395b3b2 Mon Sep 17 00:00:00 2001
From: gkroiz <gersonkroiz@google.com>
Date: Wed, 21 Feb 2024 00:21:11 +0000
Subject: [PATCH] pin main dependencies for 1.3.1 release

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

diff --git a/praxis/pip_package/requirements.txt b/praxis/pip_package/requirements.txt
index d3487e99..6640e1a8 100644
--- a/praxis/pip_package/requirements.txt
+++ b/praxis/pip_package/requirements.txt
@@ -67,7 +67,7 @@ click==8.1.7
     # via
     #   tensorflow-datasets
     #   tfds-nightly
-clu @ git+https://github.com/google/CommonLoopUtils
+clu==0.0.11
     # via -r praxis-requirements.in
 comm==0.2.1
     # via
@@ -108,11 +108,11 @@ 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
 flatbuffers==1.12
     # via tensorflow
-flax @ git+https://github.com/google/flax
+flax==0.8.1
     # via
     #   -r praxis-requirements.in
     #   clu
@@ -174,7 +174,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 praxis-requirements.in
     #   chex
diff --git a/requirements.in b/requirements.in
index d9c41a4f..b8d756c3 100644
--- a/requirements.in
+++ b/requirements.in
@@ -4,17 +4,17 @@
 absl-py
 # enforce newer chex version to avoid deprecated jax syntax
 chex>=0.1.85
-clu @ git+https://github.com/google/CommonLoopUtils
+clu==0.0.11
 einops
 etils
-fiddle @ git+https://github.com/google/fiddle
-flax @ git+https://github.com/google/flax
-jax @ git+https://github.com/google/jax
+fiddle==0.2.11
+flax==0.8.1
+jax==0.4.24
 jax-bitempered-loss
 jaxtyping
 lingvo
 numpy
-optax
+optax==0.1.9
 optax-shampoo
 opt-einsum
 # ensure sentencepiece is compataible with protobuf 3.19.6