From 25f189b2bd07f8f1ed80548532ab02b41d38e699 Mon Sep 17 00:00:00 2001 From: gkroiz Date: Fri, 16 Feb 2024 20:51:34 +0000 Subject: [PATCH] pin dependencies for 1.3.0 release --- praxis/pip_package/requirements.txt | 8 ++++---- requirements.in | 13 +++++++------ 2 files changed, 11 insertions(+), 10 deletions(-) diff --git a/praxis/pip_package/requirements.txt b/praxis/pip_package/requirements.txt index 125e6591..ad6f9cd4 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.2.24 # via # -r praxis-requirements.in # chex diff --git a/requirements.in b/requirements.in index 33b651b7..09f2838e 100644 --- a/requirements.in +++ b/requirements.in @@ -3,20 +3,21 @@ absl-py # enforce newer chex version to avoid deprecated jax syntax -chex>=0.1.85 -clu @ git+https://github.com/google/CommonLoopUtils +chex==0.1.85 +clu 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 +orbax-checkpoint==0.5.3 tensorflow-datasets==4.8.3 tensorflow-metadata==1.12.0 tensorflow-text~=2.9.0