Update praxis 1.2.0 requirements

google · Oct 16, 2023 · c0b0af8 · c0b0af8
1 parent 1a51bb8
commit c0b0af8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/praxis/pip_package/requirements.txt b/praxis/pip_package/requirements.txt
@@ -104,7 +104,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
 flatbuffers==1.12
     # via tensorflow
@@ -165,7 +165,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 praxis-requirements.in
     #   chex

diff --git a/requirements.in b/requirements.in
@@ -5,9 +5,9 @@ absl-py
 clu
 einops
 etils
-fiddle @ git+https://github.com/google/fiddle
+fiddle==0.2.11 
 flax
-jax @ git+https://github.com/google/jax
+jax==0.4.18
 jax-bitempered-loss
 jaxtyping
 lingvo