Update versions and CHANGELOG after jax 0.4.17 release

CHANGELOG.md

@@ -6,7 +6,11 @@ Best viewed [here](https://jax.readthedocs.io/en/latest/changelog.html).
 Remember to align the itemized text with the first line of an item within a list.
 -->
 
-# jax 0.4.17
+# jax 0.4.18
+
+# jaxlib 0.4.18
+
+# jax 0.4.17 (Oct 3, 2023)
 
 * New features
   * Added new {func}`jax.numpy.bitwise_count` function, matching the API of the simlar
@@ -33,7 +37,7 @@ Remember to align the itemized text with the first line of an item within a list
     Previously many of these were treated as `Any` by static type checkers like
     `mypy` and `pytype`.
 
-# jaxlib 0.4.17
+# jaxlib 0.4.17 (Oct 3, 2023)
 
 * Changes:
   * Python 3.12 wheels were added in this release.

jax/version.py

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

@@ -24,7 +24,7 @@ project_name = 'jax'
 
 _current_jaxlib_version = '0.4.17'
 # The following should be updated with each new jaxlib release.
-_latest_jaxlib_version_on_pypi = '0.4.16'
+_latest_jaxlib_version_on_pypi = '0.4.17'
 _available_cuda11_cudnn_versions = ['86']
 _default_cuda11_cudnn_version = '86'
 _default_cuda12_cudnn_version = '89'