Prepare 0.4.28 release.

CHANGELOG.md

@@ -8,16 +8,31 @@ Remember to align the itemized text with the first line of an item within a list
 
 ## jax 0.4.28
 
+* Bug fixes
+  * Reverted a change to `make_jaxpr` that was breaking Equinox (#21116).
+
 * Deprecations & removals
   * The ``kind`` argument to {func}`jax.numpy.sort` and {func}`jax.numpy.argsort`
     is now removed. Use `stable=True` or `stable=False` instead.
   * Removed ``get_compute_capability`` from the ``jax.experimental.pallas.gpu``
     module. Use the ``compute_capability`` attribute of a GPU device, returned
     by {func}`jax.devices` or {func}`jax.local_devices`, instead.
 
+* Changes
+  * The minimum jaxlib version of this release is 0.4.27.
+
 ## jaxlib 0.4.28
+
+* Bug fixes
   * Fixes a memory corruption bug in the type name of Array and JIT Python
     objects in Python 3.10 or earlier.
+  * Fixed a warning `'+ptx84' is not a recognized feature for this target`
+    under CUDA 12.4.
+  * Fixed a slow compilation problem on CPU.
+
+* Changes
+  * The Windows build is now built with Clang instead of MSVC.
+
 
 ## jax 0.4.27 (May 7, 2024)
 

setup.py

@@ -22,12 +22,12 @@
 
 project_name = 'jax'
 
-_current_jaxlib_version = '0.4.27'
+_current_jaxlib_version = '0.4.28'
 # The following should be updated with each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.4.27'
 _default_cuda12_cudnn_version = '89'
 _available_cuda12_cudnn_versions = [_default_cuda12_cudnn_version]
-_libtpu_version = '0.1.dev20240507'
+_libtpu_version = '0.1.dev20240508'
 
 def load_version_module(pkg_path):
   spec = importlib.util.spec_from_file_location(