pin to stable versions in requirements

Author: aireenmei

praxis/pip_package/requirements.txt

@@ -87,7 +87,7 @@ executing==1.2.0
     # via stack-data
 fastjsonschema==2.16.3
     # via nbformat
-fiddle @ git+https://github.com/google/fiddle
+fiddle==0.2.5
     # via -r praxis-requirements.in
 flatbuffers==1.12
     # via tensorflow
@@ -161,7 +161,7 @@ ipywidgets==8.0.6
     # via jupyter
 isoduration==20.11.0
     # via jsonschema
-jax @ git+https://github.com/google/jax
+jax==0.4.7
     # via
     #   -r praxis-requirements.in
     #   chex

requirements.in

@@ -1,16 +1,16 @@
 # To update requirements.txt for praxis and paxml, run:
 # cd ../../paxml/pip_package && bash ./compile_requirements.sh
 
-absl-py
-clu
-einops
-fiddle @ git+https://github.com/google/fiddle
-flax
-jax @ git+https://github.com/google/jax
-jax-bitempered-loss
-lingvo
-numpy
-optax
-optax-shampoo
-tensorflow~=2.9.2
-tensorflow-text~=2.9.0
+absl-py==1.4.0
+clu==0.0.8
+einops==0.6.0
+fiddle==0.2.5
+flax==0.6.8
+jax==0.4.7
+jax-bitempered-loss==0.0.2
+lingvo==0.12.6
+numpy~=1.24.2
+optax==0.1.4
+optax-shampoo==0.0.6
+tensorflow==2.9.3
+tensorflow-text==2.9.0