diff --git a/paxml/pip_package/requirements.txt b/paxml/pip_package/requirements.txt index d73c76585..a75252ec0 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.3.0 # 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.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 diff --git a/requirements.in b/requirements.in index 1d3fb07a3..3dfb9108a 100644 --- a/requirements.in +++ b/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