Merge branch 'release/0.4.37' into main.

CHANGELOG.md

@@ -10,12 +10,18 @@ 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.
 -->
 
-## jax 0.4.37
+## jax 0.4.38
+
+## jax 0.4.37 (Dec 9, 2024)
+
+This is a patch release of jax 0.4.36. Only "jax" was released at this version.
 
 * Bug fixes
+  * Fixed a bug where `jit` would error if an argument was named `f` (#25329).
   * Fix a bug that will throw `index out of range` error in
     {func}`jax.lax.while_loop` if the user register pytree node class with
     different aux data for the flatten and flatten_with_path.
+  * Pinned a new libtpu release (0.0.6) that fixes a compiler bug on TPU v6e.
 
 ## jax 0.4.36 (Dec 5, 2024)
 

jax/version.py

@@ -21,7 +21,7 @@ import os
 import pathlib
 import subprocess
 
-_version = "0.4.37"
+_version = "0.4.38"
 # 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

@@ -23,7 +23,7 @@ _current_jaxlib_version = '0.4.36'
 # The following should be updated after each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.4.36'
 
-_libtpu_version = '0.0.5'
+_libtpu_version = '0.0.6'
 _libtpu_nightly_terminal_version = '0.1.dev20241010+nightly.cleanup'
 
 def load_version_module(pkg_path):