From c161c6287821053dbe051c31a8d26f9478717d78 Mon Sep 17 00:00:00 2001
From: Yash Katariya <yashkatariya@google.com>
Date: Fri, 18 Feb 2022 16:23:02 -0800
Subject: [PATCH] Finish jax release

PiperOrigin-RevId: 429670894
---
 CHANGELOG.md   | 14 +++++++++-----
 jax/version.py |  2 +-
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 301dd270d..146808d9c 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -8,9 +8,16 @@ Remember to align the itemized text with the first line of an item within a list
 PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
 -->
 
-## jax 0.3.1 (Unreleased)
+## jax 0.3.2 (Unreleased)
 * [GitHub
-  commits](https://github.com/google/jax/compare/jax-v0.3.0...main).
+  commits](https://github.com/google/jax/compare/jax-v0.3.1...main).
+
+## jaxlib 0.3.1 (Unreleased)
+* Changes
+
+## jax 0.3.1 (Feb 18, 2022)
+* [GitHub
+  commits](https://github.com/google/jax/compare/jax-v0.3.0...jax-v0.3.1).
 
 * Changes:
   * `jax.test_util.JaxTestCase` and `jax.test_util.JaxTestLoader` are now deprecated.
@@ -30,9 +37,6 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
     {func}`jax.scipy.signal.csd`, {func}`jax.scipy.signal.stft`,
     {func}`jax.scipy.signal.welch`.
 
-## jaxlib 0.3.1 (Unreleased)
-* Changes
-
 
 ## jax 0.3.0 (Feb 10, 2022)
 * [GitHub
diff --git a/jax/version.py b/jax/version.py
index ba0e675e8..0a501afcc 100644
--- a/jax/version.py
+++ b/jax/version.py
@@ -12,6 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "0.3.1"
+__version__ = "0.3.2"
 
 _minimum_jaxlib_version = "0.1.74"