Merge pull request #17989 from skye:version

PiperOrigin-RevId: 571418823

Author: jax authors

CHANGELOG.md

@@ -26,6 +26,9 @@ Remember to align the itemized text with the first line of an item within a list
       `standard_primitive`, `standard_unop`, `unop`, and `unop_dtype_rule` are
       internal utilities, now deprecated without replacement.
 
+* Bug fixes
+  * Fixed Cloud TPU regression where compilation would OOM due to smem.
+
 # jax 0.4.17 (Oct 3, 2023)
 
 * New features

setup.py

@@ -22,13 +22,13 @@
 
 project_name = 'jax'
 
-_current_jaxlib_version = '0.4.17'
+_current_jaxlib_version = '0.4.18'
 # The following should be updated with each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.4.17'
 _available_cuda11_cudnn_versions = ['86']
 _default_cuda11_cudnn_version = '86'
 _default_cuda12_cudnn_version = '89'
-_libtpu_version = '0.1.dev20231003'
+_libtpu_version = '0.1.dev20231006'
 
 def load_version_module(pkg_path):
   spec = importlib.util.spec_from_file_location(