Merge pull request #17989 from skye:version

PiperOrigin-RevId: 571418823
google · Oct 6, 2023 · e27af48 · e27af48
2 parents f0e4ea2 + d4a1bb9
commit e27af48
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/CHANGELOG.md b/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

diff --git a/setup.py b/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(