Pin paxml requirements

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,15 +128,15 @@ executing==2.0.1
     # via stack-data
 fastjsonschema==2.19.1
     # via nbformat
-fiddle @ git+https://github.com/google/fiddle
+fiddle==0.3.0
     # via -r praxis-requirements.in
 filelock==3.13.1
     # via
     #   huggingface-hub
     #   transformers
 flatbuffers==1.12
     # via tensorflow
-flax @ git+https://github.com/google/flax
+flax==0.8.2
     # 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.26
     # via
     #   -r paxml-requirements.in
     #   -r praxis-requirements.in
@@ -433,7 +433,7 @@ optax==0.2.1
     #   flax
 optax-shampoo==0.0.6
     # via -r praxis-requirements.in
-orbax-checkpoint==0.5.6
+orbax-checkpoint==0.5.9
     # via
     #   -r paxml-requirements.in
     #   flax

requirements.in

@@ -1,23 +1,23 @@
 # To update requirements.txt for praxis and paxml, run:
 # bash ./compile_requirements.sh
 
-absl-py
-clu @ git+https://github.com/google/CommonLoopUtils
-etils
-flax @ git+https://github.com/google/flax
-graphviz
-jax @ git+https://github.com/google/jax
-lingvo
-numpy
-orbax-checkpoint
-praxis
+absl-py==1.4.0
+clu==0.0.11
+etils==1.7.0
+flax==0.8.2
+graphviz==0.20.1
+jax==0.4.26
+lingvo==0.12.7
+numpy==1.26.4
+orbax-checkpoint==0.5.9
+praxis==1.4.0
 protobuf==3.19.6
-pyglove
-seqio-nightly
-t5
+pyglove==0.4.4
+seqio-nightly==0.0.17.dev20231010
+t5==0.9.4
 tensorflow~=2.9.2
 tensorflow-text~=2.9.0
-tensorstore
+tensorstore==0.1.55
 tensorflow-datasets==4.8.3
 tfds-nightly==4.8.3.dev202303280045
 tensorflow-metadata==1.12.0