Merge pull request jax-ml#4573 from google:update-pypi

PiperOrigin-RevId: 337028688
thuang513 · Oct 14, 2020 · 6d95b6c · 6d95b6c
2 parents cb4a014 + 503bc36
commit 6d95b6c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/docs/CHANGELOG.rst b/docs/CHANGELOG.rst
@@ -9,6 +9,10 @@ Change Log
 
 These are the release notes for JAX.
 
+jax 0.2.2 (October 13 2020)
+--------------------------
+* `GitHub commits <https://github.com/google/jax/compare/jax-v0.2.1...jax-v0.2.2>`_.
+
 jax 0.2.1 (October 6 2020)
 --------------------------
 * `GitHub commits <https://github.com/google/jax/compare/jax-v0.2.0...jax-v0.2.1>`_.

diff --git a/jax/interpreters/xla.py b/jax/interpreters/xla.py
@@ -566,7 +566,7 @@ def _xla_call_impl(fun: lu.WrappedFun, *args, device, backend, name, donated_inv
     # In addition to re-executing the Python code, which is usually undesirable
     # but which FLAGS.jax_debug_nans is meant to opt into, we'll be re-executing
     # any linear_util.py-style side effects, i.e. re-populating Stores created
-    # by any transformation_with(aux's applied to fun. Since this is
+    # by any transformation_with_aux's applied to fun. Since this is
     # intentional here, to avoid "Store occupied" errors we reset the stores to
     # be empty.
     for store in fun.stores: store.reset()

diff --git a/jax/version.py b/jax/version.py
@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "0.2.1"
+__version__ = "0.2.2"