Finish jax and jaxlib 0.4.28 release

PiperOrigin-RevId: 632653310

CHANGELOG.md

@@ -6,7 +6,11 @@ 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.28
+## jax 0.4.29
+
+## jaxlib 0.4.29
+
+## jax 0.4.28 (May 9, 2024)
 
 * Bug fixes
   * Reverted a change to `make_jaxpr` that was breaking Equinox (#21116).
@@ -23,7 +27,7 @@ Remember to align the itemized text with the first line of an item within a list
 * Changes
   * The minimum jaxlib version of this release is 0.4.27.
 
-## jaxlib 0.4.28
+## jaxlib 0.4.28 (May 9, 2024)
 
 * Bug fixes
   * Fixes a memory corruption bug in the type name of Array and JIT Python

jax/version.py

@@ -21,7 +21,7 @@
 import pathlib
 import subprocess
 
-_version = "0.4.28"
+_version = "0.4.29"
 # The following line is overwritten by build scripts in distributions &
 # releases. Do not modify this manually, or jax/jaxlib build will fail.
 _release_version: str | None = None

setup.py

@@ -24,7 +24,7 @@
 
 _current_jaxlib_version = '0.4.28'
 # The following should be updated with each new jaxlib release.
-_latest_jaxlib_version_on_pypi = '0.4.27'
+_latest_jaxlib_version_on_pypi = '0.4.28'
 _default_cuda12_cudnn_version = '89'
 _available_cuda12_cudnn_versions = [_default_cuda12_cudnn_version]
 _libtpu_version = '0.1.dev20240508'