Release JAX 0.5.0

Author: hawkinsp

CHANGELOG.md

@@ -14,18 +14,33 @@ Remember to align the itemized text with the first line of an item within a list
 When releasing, please add the new-release-boilerplate to docs/pallas/CHANGELOG.md.
 -->
 
-## Unreleased
+## jax 0.5.0 (Jan 17, 2025)
 
 As of this release, JAX now uses
 [effort-based versioning](https://jax.readthedocs.io/en/latest/jep/25516-effver.html).
-Since this release makes a small but breaking change to PRNG key semantics that
+Since this release makes a breaking change to PRNG key semantics that
 may require users to update their code, we are bumping the "meso" version of JAX
 to signify this.
 
 * Breaking changes
   * Enable `jax_threefry_partitionable` by default (see
     [the update note](https://github.com/jax-ml/jax/discussions/18480)).
 
+  * This release drops support for Mac x86 wheels. Mac ARM of course remains
+    supported. For a recent discussion, see
+    https://github.com/jax-ml/jax/discussions/22936.
+
+    Two key factors motivated this decision:
+    * The Mac x86 build (only) has a number of test failures and crashes. We
+      would prefer to ship no release than a broken release.
+    * Mac x86 hardware is end-of-life and cannot be easily obtained for
+      developers at this point. So it is difficult for us to fix this kind of
+      problem even if we wanted to.
+
+    We are open to readding support for Mac x86 if the community is willing
+    to help support that platform: in particular, we would need the JAX test
+    suite to pass cleanly on Mac x86 before we could ship releases again.
+
 * Changes:
   * The minimum NumPy version is now 1.25. NumPy 1.25 will remain the minimum
     supported version until June 2025.
@@ -47,6 +62,7 @@ to signify this.
     {func}`jax.ffi.register_ffi_type_id` function.
   * The AOT lowering `.as_text()` method now supports the `debug_info` option
     to include debugging information, e.g., source location, in the output.
+
 * Deprecations
   * From {mod}`jax.interpreters.xla`, `abstractify` and `pytype_aval_mappings`
     are now deprecated, having been replaced by symbols of the same name

jax/version.py

@@ -137,7 +137,7 @@ def make_release_tree(self, base_dir, files):
 
 
 __version__ = _get_version_string()
-_minimum_jaxlib_version = "0.4.38"
+_minimum_jaxlib_version = "0.5.0"
 
 def _version_as_tuple(version_str):
   return tuple(int(i) for i in version_str.split(".") if i.isdigit())

setup.py

@@ -19,11 +19,11 @@
 
 project_name = 'jax'
 
-_current_jaxlib_version = '0.4.38'
+_current_jaxlib_version = '0.5.0'
 # The following should be updated after each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.4.38'
 
-_libtpu_version = '0.0.7'
+_libtpu_version = '0.0.8'
 _libtpu_nightly_terminal_version = '0.1.dev20241010+nightly.cleanup'
 
 def load_version_module(pkg_path):