praxis and paxml 0.2.1 release

PiperOrigin-RevId: 490026727

RELEASE.md

@@ -1,3 +1,8 @@
+# Version: 0.2.1
+## Note
+*   Version: 0.2.1
+*   Build Date: 20221121
+*   Praxis commit: f7e98026c1c5ecbc6e4aff175621d443fa37fcf2
 # Version: 0.2.0
 ## Major Features and Improvements
 *  Preparatory work for Fiddle integration

praxis/pip_package/cloudbuild-release.yaml

@@ -16,7 +16,7 @@ steps:
 
 substitutions:
     _PYTHON_VERSION: '3.8'
-    _RELEASE_VERSION: '0.2.0'  # or rX.Y
+    _RELEASE_VERSION: '0.2.1'  # or rX.Y
     _IMAGE_NAME: 'praxis_${_RELEASE_VERSION}_${_PYTHON_VERSION}'
     _WHEEL_FOLDER: '/tmp/wheels'
 options:

praxis/pip_package/requirements.txt

@@ -53,6 +53,8 @@ chex==0.1.5
     # via optax
 clu==0.0.7
     # via -r praxis-requirements.in
+comm==0.1.0
+    # via ipykernel
 commonmark==0.9.1
     # via rich
 contextlib2==21.6.0
@@ -126,7 +128,7 @@ importlib-resources==5.10.0
     #   etils
     #   jsonschema
     #   tensorflow-datasets
-ipykernel==6.17.1
+ipykernel==6.18.0
     # via
     #   ipywidgets
     #   jupyter
@@ -195,7 +197,7 @@ jupyter-core==5.0.0
     #   qtconsole
 jupyter-http-over-ws==0.0.8
     # via lingvo
-jupyter-server==1.23.2
+jupyter-server==1.23.3
     # via
     #   nbclassic
     #   notebook-shim
@@ -272,7 +274,7 @@ notebook==6.5.2
     #   jupyter-http-over-ws
 notebook-shim==0.2.2
     # via nbclassic
-numpy==1.23.4
+numpy==1.23.5
     # via
     #   -r praxis-requirements.in
     #   chex
@@ -298,7 +300,7 @@ opt-einsum==3.3.0
     # via
     #   jax
     #   tensorflow
-optax==0.1.3
+optax==0.1.4
     # via
     #   -r praxis-requirements.in
     #   flax
@@ -336,7 +338,7 @@ prometheus-client==0.15.0
     #   notebook
 promise==2.3
     # via tensorflow-datasets
-prompt-toolkit==3.0.32
+prompt-toolkit==3.0.33
     # via
     #   ipython
     #   jupyter-console
@@ -481,7 +483,7 @@ tensorflow-text==2.9.0
     #   lingvo
 tensorstore==0.1.28
     # via flax
-termcolor==2.1.0
+termcolor==2.1.1
     # via
     #   tensorflow
     #   tensorflow-datasets
@@ -509,6 +511,7 @@ tqdm==4.64.1
     # via tensorflow-datasets
 traitlets==5.5.0
     # via
+    #   comm
     #   ipykernel
     #   ipython
     #   ipywidgets

praxis/pip_package/setup.py

@@ -39,7 +39,7 @@ def _get_requirements():
 
 setup(
     name='praxis',
-    version='0.2.0',
+    version='0.2.1',
     description=(
         'Functionalities such as a layers for building neural networks in Jax.'
     ),