Release stable paxml/praxis 1.1.0

google · Aug 25, 2023 · 94dc6f7 · 94dc6f7
1 parent d0454bf
commit 94dc6f7
Show file tree

Hide file tree

Showing 3 changed files with 64 additions and 62 deletions.
diff --git a/RELEASE.md b/RELEASE.md
@@ -6,8 +6,8 @@
 * Add support for Direct Preference Optimization (DPO).
 ## Note
 *   Version: 1.1.0
-*   Build Date: 20230712
-*   Praxis commit: f457672aa6193621a4ebf2f7d9c9acf62e0513f6
+*   Build Date: 20230824
+*   Praxis commit: 2a7d0407871502a1d79dcd0e01411e73f1d15d36
 # Version: 1.0.0
 ## Major Features and Improvements
 * **Fiddle** - Praxis layers and BaseParameterizable are now configured with [Fiddle](https://github.com/google/fiddle), a Python-first configuration library. Fiddle reduces boilerplate, and adds productivity features including history tracking, graphviz visualization, support for aliasing objects, and more.

diff --git a/praxis/pip_package/requirements.txt b/praxis/pip_package/requirements.txt
@@ -2,11 +2,11 @@
 # This file is autogenerated by pip-compile with Python 3.10
 # by the following command:
 #
-#    pip-compile --output-file=praxis-requirements.txt praxis-requirements.in
+#    pip-compile --output-file=requirements.txt requirements.in
 #
 absl-py==1.4.0
     # via
-    #   -r praxis-requirements.in
+    #   -r requirements.in
     #   array-record
     #   chex
     #   clu
@@ -21,11 +21,11 @@ absl-py==1.4.0
     #   tfds-nightly
 anyio==3.7.1
     # via jupyter-server
-argon2-cffi==21.3.0
+argon2-cffi==23.1.0
     # via jupyter-server
 argon2-cffi-bindings==21.2.0
     # via argon2-cffi
-array-record==0.4.0
+array-record==0.4.1
     # via tfds-nightly
 arrow==1.2.3
     # via isoduration
@@ -58,12 +58,12 @@ charset-normalizer==3.2.0
     # via requests
 chex==0.1.82
     # via optax
-click==8.1.6
+click==8.1.7
     # via
     #   tensorflow-datasets
     #   tfds-nightly
 clu==0.0.9
-    # via -r praxis-requirements.in
+    # via -r requirements.in
 comm==0.1.4
     # via
     #   ipykernel
@@ -74,7 +74,7 @@ contourpy==1.1.0
     # via matplotlib
 cycler==0.11.0
     # via matplotlib
-debugpy==1.6.7
+debugpy==1.6.7.post1
     # via ipykernel
 decorator==5.1.1
     # via ipython
@@ -85,31 +85,31 @@ dm-tree==0.1.8
     #   tensorflow-datasets
     #   tfds-nightly
 einops==0.6.1
-    # via -r praxis-requirements.in
+    # via -r requirements.in
 etils[enp,epath,epy]==1.4.1
     # via
-    #   -r praxis-requirements.in
+    #   -r requirements.in
     #   array-record
     #   clu
     #   lingvo
     #   orbax-checkpoint
     #   tensorflow-datasets
     #   tfds-nightly
-exceptiongroup==1.1.2
+exceptiongroup==1.1.3
     # via anyio
 executing==1.2.0
     # via stack-data
 fastjsonschema==2.18.0
     # via nbformat
-fiddle @ git+https://github.com/google/fiddle
-    # via -r praxis-requirements.in
+fiddle==0.2.10
+    # via -r requirements.in
 flatbuffers==1.12
     # via tensorflow
-flax==0.7.1
+flax==0.7.2
     # via
-    #   -r praxis-requirements.in
+    #   -r requirements.in
     #   clu
-fonttools==4.42.0
+fonttools==4.42.1
     # via matplotlib
 fqdn==1.5.1
     # via jsonschema
@@ -129,7 +129,7 @@ graph-compression-google-research==0.0.4
     # via lingvo
 graphviz==0.20.1
     # via fiddle
-grpcio==1.56.2
+grpcio==1.57.0
     # via
     #   tensorboard
     #   tensorflow
@@ -160,24 +160,24 @@ ipywidgets==8.1.0
     # via jupyter
 isoduration==20.11.0
     # via jsonschema
-jax @ git+https://github.com/google/jax
+jax==0.4.14
     # via
-    #   -r praxis-requirements.in
+    #   -r requirements.in
     #   chex
     #   clu
     #   flax
     #   optax
     #   orbax-checkpoint
 jax-bitempered-loss==0.0.2
-    # via -r praxis-requirements.in
+    # via -r requirements.in
 jaxlib==0.4.14
     # via
     #   chex
     #   clu
     #   optax
     #   orbax-checkpoint
-jaxtyping==0.2.20
-    # via -r praxis-requirements.in
+jaxtyping==0.2.21
+    # via -r requirements.in
 jedi==0.19.0
     # via ipython
 jinja2==3.1.2
@@ -186,7 +186,7 @@ jinja2==3.1.2
     #   jupyterlab
     #   jupyterlab-server
     #   nbconvert
-joblib==1.3.1
+joblib==1.3.2
     # via scikit-learn
 json5==0.9.14
     # via jupyterlab-server
@@ -227,7 +227,7 @@ jupyter-http-over-ws==0.0.8
     # via lingvo
 jupyter-lsp==2.2.0
     # via jupyterlab
-jupyter-server==2.7.0
+jupyter-server==2.7.2
     # via
     #   jupyter-lsp
     #   jupyterlab
@@ -236,7 +236,7 @@ jupyter-server==2.7.0
     #   notebook-shim
 jupyter-server-terminals==0.4.4
     # via jupyter-server
-jupyterlab==4.0.4
+jupyterlab==4.0.5
     # via notebook
 jupyterlab-pygments==0.2.2
     # via nbconvert
@@ -250,14 +250,14 @@ keras==2.9.0
     # via tensorflow
 keras-preprocessing==1.1.2
     # via tensorflow
-kiwisolver==1.4.4
+kiwisolver==1.4.5
     # via matplotlib
 libclang==16.0.6
     # via tensorflow
 libcst==1.0.1
     # via fiddle
 lingvo==0.12.7
-    # via -r praxis-requirements.in
+    # via -r requirements.in
 markdown==3.4.4
     # via tensorboard
 markdown-it-py==3.0.0
@@ -295,7 +295,7 @@ mypy-extensions==1.0.0
     # via typing-inspect
 nbclient==0.8.0
     # via nbconvert
-nbconvert==7.7.3
+nbconvert==7.7.4
     # via
     #   jupyter
     #   jupyter-server
@@ -318,7 +318,7 @@ notebook-shim==0.2.3
     #   notebook
 numpy==1.25.2
     # via
-    #   -r praxis-requirements.in
+    #   -r requirements.in
     #   chex
     #   clu
     #   contourpy
@@ -350,11 +350,11 @@ opt-einsum==3.3.0
     #   tensorflow
 optax==0.1.7
     # via
-    #   -r praxis-requirements.in
+    #   -r requirements.in
     #   flax
 optax-shampoo==0.0.6
-    # via -r praxis-requirements.in
-orbax-checkpoint==0.3.2
+    # via -r requirements.in
+orbax-checkpoint==0.3.5
     # via flax
 overrides==7.4.0
     # via jupyter-server
@@ -447,7 +447,7 @@ pyyaml==6.0.1
     #   libcst
     #   ml-collections
     #   orbax-checkpoint
-pyzmq==25.1.0
+pyzmq==25.1.1
     # via
     #   ipykernel
     #   jupyter-client
@@ -490,7 +490,7 @@ rsa==4.9
     # via google-auth
 scikit-learn==1.3.0
     # via lingvo
-scipy==1.11.1
+scipy==1.11.2
     # via
     #   jax
     #   jaxlib
@@ -529,12 +529,12 @@ tensorboard-plugin-wit==1.8.1
     # via tensorboard
 tensorflow==2.9.3
     # via
-    #   -r praxis-requirements.in
+    #   -r requirements.in
     #   lingvo
     #   tensorflow-text
 tensorflow-datasets==4.8.3
     # via
-    #   -r praxis-requirements.in
+    #   -r requirements.in
     #   lingvo
 tensorflow-estimator==2.9.0
     # via tensorflow
@@ -546,14 +546,14 @@ tensorflow-io-gcs-filesystem==0.33.0
     # via tensorflow
 tensorflow-metadata==1.12.0
     # via
-    #   -r praxis-requirements.in
+    #   -r requirements.in
     #   tensorflow-datasets
     #   tfds-nightly
 tensorflow-text==2.9.0
     # via
-    #   -r praxis-requirements.in
+    #   -r requirements.in
     #   lingvo
-tensorstore==0.1.40
+tensorstore==0.1.41
     # via
     #   flax
     #   orbax-checkpoint
@@ -567,7 +567,7 @@ terminado==0.17.1
     #   jupyter-server
     #   jupyter-server-terminals
 tfds-nightly==4.8.3.dev202303280045
-    # via -r praxis-requirements.in
+    # via -r requirements.in
 threadpoolctl==3.2.0
     # via scikit-learn
 tinycss2==1.2.1
@@ -580,7 +580,7 @@ tomli==2.0.1
     # via jupyterlab
 toolz==0.12.0
     # via chex
-tornado==6.3.2
+tornado==6.3.3
     # via
     #   ipykernel
     #   jupyter-client
@@ -589,7 +589,7 @@ tornado==6.3.2
     #   jupyterlab
     #   notebook
     #   terminado
-tqdm==4.65.0
+tqdm==4.66.1
     # via
     #   tensorflow-datasets
     #   tfds-nightly
@@ -610,8 +610,10 @@ traitlets==5.9.0
     #   nbconvert
     #   nbformat
     #   qtconsole
-typeguard==4.1.0
-    # via jaxtyping
+typeguard==4.1.2
+    # via
+    #   -r requirements.in
+    #   jaxtyping
 typing-extensions==4.7.1
     # via
     #   async-lru
@@ -642,11 +644,11 @@ webencodings==0.5.1
     # via
     #   bleach
     #   tinycss2
-websocket-client==1.6.1
+websocket-client==1.6.2
     # via jupyter-server
-werkzeug==2.3.6
+werkzeug==2.3.7
     # via tensorboard
-wheel==0.41.1
+wheel==0.41.2
     # via
     #   astunparse
     #   tensorboard

diff --git a/requirements.in b/requirements.in
@@ -1,22 +1,22 @@
 # To update requirements.txt for praxis and paxml, run:
 # cd ../../paxml/pip_package && bash ./compile_requirements.sh
 
-absl-py
-clu
-einops
-etils
-fiddle @ git+https://github.com/google/fiddle
-flax
-jax @ git+https://github.com/google/jax
-jax-bitempered-loss
-jaxtyping
-lingvo
-numpy
-optax
-optax-shampoo
+absl-py==1.4.0
+clu==0.0.9
+einops==0.6.1
+etils[enp,epath,epy]==1.4.1
+fiddle==0.2.10
+flax==0.7.2
+jax==0.4.14
+jax-bitempered-loss==0.0.2
+jaxtyping==0.2.21
+lingvo==0.12.7
+numpy==1.25.2
+optax==0.1.7
+optax-shampoo==0.0.6
 tensorflow-datasets==4.8.3
 tensorflow-metadata==1.12.0
 tensorflow-text~=2.9.0
 tensorflow~=2.9.2
 tfds-nightly==4.8.3.dev202303280045
-typeguard
+typeguard==4.1.2