Update Pax 1.2.0 requirements

google · Oct 17, 2023 · e57709b · e57709b
1 parent fc56bde
commit e57709b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/paxml/pip_package/requirements.txt b/paxml/pip_package/requirements.txt
@@ -124,7 +124,7 @@ executing==2.0.0
     # via stack-data
 fastjsonschema==2.18.1
     # via nbformat
-fiddle @ git+https://github.com/google/fiddle
+fiddle==0.2.11
     # via -r praxis-requirements.in
 filelock==3.12.4
     # via
@@ -206,7 +206,7 @@ ipywidgets==8.1.1
     # via jupyter
 isoduration==20.11.0
     # via jsonschema
-jax @ git+https://github.com/google/jax
+jax==0.4.18
     # via
     #   -r paxml-requirements.in
     #   -r praxis-requirements.in

diff --git a/requirements.in b/requirements.in
@@ -6,7 +6,7 @@ clu
 etils
 flax
 graphviz
-jax @ git+https://github.com/google/jax
+jax==0.4.18
 lingvo
 numpy
 orbax-checkpoint
@@ -20,4 +20,4 @@ tensorflow-text~=2.9.0
 tensorstore
 tensorflow-datasets==4.8.3
 tfds-nightly==4.8.3.dev202303280045
-tensorflow-metadata==1.12.0
+tensorflow-metadata==1.12.0