1
0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 21:36:05 +00:00

Finish jax and jaxlib 0.4.7 release

PiperOrigin-RevId: 519839723
This commit is contained in:
Yash Katariya 2023-03-27 15:06:06 -07:00 committed by jax authors
parent 10dc941d8d
commit 670fba3a91
3 changed files with 8 additions and 4 deletions

@ -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.7
## jax 0.4.8
## jaxlib 0.4.8
## jax 0.4.7 (March 27, 2023)
* Changes
* As per https://jax.readthedocs.io/en/latest/jax_array_migration.html#jax-array-migration
@ -39,7 +43,7 @@ Remember to align the itemized text with the first line of an item within a list
jax.Arrays as input and remove the `in_shardings` argument to pjit since
it is optional.
## jaxlib 0.4.7
## jaxlib 0.4.7 (March 27, 2023)
Changes:
* jaxlib now depends on `ml_dtypes`, which contains definitions of NumPy types

@ -15,7 +15,7 @@
# This file is included as part of both jax and jaxlib. It is also
# eval()-ed by setup.py, so it should not have any dependencies.
__version__ = "0.4.7"
__version__ = "0.4.8"
_minimum_jaxlib_version = "0.4.6"
def _version_as_tuple(version_str):

@ -21,7 +21,7 @@ from setuptools import setup, find_packages
_current_jaxlib_version = '0.4.7'
# The following should be updated with each new jaxlib release.
_latest_jaxlib_version_on_pypi = '0.4.6'
_latest_jaxlib_version_on_pypi = '0.4.7'
_available_cuda11_cudnn_versions = ['82', '86']
_default_cuda11_cudnn_version = '86'
_default_cuda12_cudnn_version = '88'