Merge pull request #14357 from skye:version

PiperOrigin-RevId: 508119911

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)
 

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):

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']