Release jax and jaxlib 0.3.0 as per the new release process.

PiperOrigin-RevId: 427809845

CHANGELOG.md

@@ -8,22 +8,28 @@ Remember to align the itemized text with the first line of an item within a list
 PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
 -->
 
-## jax 0.2.29 (Unreleased)
+## jax 0.3.0 (Feb 10, 2022)
 * [GitHub
   commits](https://github.com/google/jax/compare/jax-v0.2.28...main).
 
-## jax 0.2.28 (Feb 1 2022)
+* Changes
+  * jax version has been bumped to 0.3.0. Please see the [design doc](https://jax.readthedocs.io/en/latest/design_notes/jax_versioning.html)
+    for the explanation.
+
+## jaxlib 0.3.0 (Feb 10, 2022)
+* Changes
+  * Bazel 5.0.0 is now required to build jaxlib.
+  * jaxlib version has been bumped to 0.3.0. Please see the [design doc](https://jax.readthedocs.io/en/latest/design_notes/jax_versioning.html)
+    for the explanation.
+
+## jax 0.2.28 (Feb 1, 2022)
 * [GitHub
-  commits](https://github.com/google/jax/compare/jax-v0.2.27...main).
+  commits](https://github.com/google/jax/compare/jax-v0.2.27...jax-v0.2.28).
   * `jax.jit(f).lower(...).compiler_ir()` now defaults to the MHLO dialect if no
     `dialect=` is passed.
   * The `jax.jit(f).lower(...).compiler_ir(dialect='mhlo')` now returns an MLIR
     `ir.Module` object instead of its string representation.
 
-## jaxlib 0.1.77 (Unreleased)
-* Changes
-  * Bazel 5.0.0 is now required to build jaxlib.
-
 ## jaxlib 0.1.76 (Jan 27, 2022)
 
 * New features

docs/requirements.txt

@@ -20,4 +20,4 @@ pytest-xdist
 matplotlib
 scikit-learn
 numpy
-.[cpu]  # Install jax from the current directory; jaxlib from pypi.
+.[ci]  # Install jax from the current directory; jaxlib from pypi.

jax/version.py

@@ -12,6 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "0.2.29"
+__version__ = "0.3.0"
 
 _minimum_jaxlib_version = "0.1.74"

jaxlib/version.py

@@ -17,4 +17,4 @@
 # reflect the most recent available binaries.
 # __version__ should be increased after releasing the current version
 # (i.e. on main, this is always the next version to be released).
-__version__ = "0.1.77"
+__version__ = "0.3.0"

setup.py

@@ -14,7 +14,7 @@
 
 from setuptools import setup, find_packages
 
-_current_jaxlib_version = '0.1.76'
+_current_jaxlib_version = '0.3.0'
 # The following should be updated with each new jaxlib release.
 _latest_jaxlib_version_on_pypi = '0.1.76'
 _available_cuda_versions = ['11']
@@ -54,7 +54,7 @@
         'cpu': [f'jaxlib=={_current_jaxlib_version}'],
 
         # Used only for CI builds that install JAX from github HEAD.
-        '_ci': [f'jaxlib=={_latest_jaxlib_version_on_pypi}'],
+        'ci': [f'jaxlib=={_latest_jaxlib_version_on_pypi}'],
 
         # Cloud TPU VM jaxlib can be installed via:
         # $ pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/jax_releases.html