From 21f12183bf35380c522fc17a84844e40fd3a22f2 Mon Sep 17 00:00:00 2001
From: Skye Wanderman-Milne <skyewm@google.com>
Date: Wed, 8 Feb 2023 10:08:57 -0800
Subject: [PATCH] Post 0.4.3 release updates

---
 CHANGELOG.md   | 10 +++++++---
 jax/version.py |  2 +-
 setup.py       |  2 +-
 3 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 3232b25ad..b0da0491e 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -6,15 +6,19 @@ Best viewed [here](https://jax.readthedocs.io/en/latest/changelog.html).
 Remember to align the itemized text with the first line of an item within a list.
 -->
 
-## jax 0.4.3
+## jax 0.4.4
+
+## jaxlib 0.4.4
+
+## jax 0.4.3 (Feb 8, 2023)
   * Breaking changes
-    * Deleted {func}`jax.scipy.linalg.polar_unitary`, which was a deprecated JAX 
+    * Deleted {func}`jax.scipy.linalg.polar_unitary`, which was a deprecated JAX
       extension to the scipy API. Use {func}`jax.scipy.linalg.polar` instead.
 
   * Changes
     * Added {func}`jax.scipy.stats.rankdata`.
 
-## jaxlib 0.4.3
+## jaxlib 0.4.3 (Feb 8, 2023)
 
 ## jax 0.4.2 (Jan 24, 2023)
 
diff --git a/jax/version.py b/jax/version.py
index 8a68db5e8..28330f929 100644
--- a/jax/version.py
+++ b/jax/version.py
@@ -15,7 +15,7 @@
 # This file is included as part of both jax and jaxlib. It is also
 # eval()-ed by setup.py, so it should not have any dependencies.
 
-__version__ = "0.4.3"
+__version__ = "0.4.4"
 _minimum_jaxlib_version = "0.4.2"
 
 def _version_as_tuple(version_str):
diff --git a/setup.py b/setup.py
index ce20b1455..4bccce9cb 100644
--- a/setup.py
+++ b/setup.py
@@ -21,7 +21,7 @@ from setuptools import setup, find_packages
 
 _current_jaxlib_version = '0.4.3'
 # The following should be updated with each new jaxlib release.
-_latest_jaxlib_version_on_pypi = '0.4.2'
+_latest_jaxlib_version_on_pypi = '0.4.3'
 _available_cuda_versions = ['11']
 _default_cuda_version = '11'
 _available_cudnn_versions = ['82', '86']