Prepare for JAX release 0.8.1

Author: danielsuo

CHANGELOG.md

@@ -14,7 +14,7 @@ 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.8.1
 
 * New features:
 

docs/pallas/CHANGELOG.md

@@ -11,7 +11,7 @@ For the overall JAX change log see [here](https://docs.jax.dev/en/latest/changel
 Remember to align the itemized text with the first line of an item within a list.
 -->
 
-## Unreleased
+## Released with jax 0.8.1
 
 * New features:
 

jax/version.py

@@ -152,7 +152,7 @@ def make_release_tree(self, base_dir, files):
 
 
 __version__ = _get_version_string()
-_minimum_jaxlib_version = '0.8.0'
+_minimum_jaxlib_version = '0.8.1'
 
 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.8.0'
+_current_jaxlib_version = '0.8.1'
 # The following should be updated after each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.8.0'
 
-_libtpu_version = '0.0.24.*'
+_libtpu_version = '0.0.30.*'
 
 def load_version_module(pkg_path):
   spec = importlib.util.spec_from_file_location(