mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 04:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Finish jax and jaxlib 0.4.22 release

Browse Source 
PiperOrigin-RevId: 590775311
This commit is contained in:
Yash Katariya 2023-12-13 18:26:06 -08:00 committed by jax authors
parent 3380b9feee
commit 25c16c0b78
3 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
CHANGELOG.md
View File

@ -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. Remember to align the itemized text with the first line of an item within a list.
--> -->
## jax 0.4.22 ## jax 0.4.23
## jaxlib 0.4.23
## jax 0.4.22 (Dec 13, 2023)
* Deprecations * Deprecations
* The `device_buffer` and `device_buffers` properties of JAX arrays are deprecated. * The `device_buffer` and `device_buffers` properties of JAX arrays are deprecated.
@ -15,7 +19,7 @@ Remember to align the itemized text with the first line of an item within a list
* `arr.device_buffer` becomes `arr.addressable_data(0)` * `arr.device_buffer` becomes `arr.addressable_data(0)`
* `arr.device_buffers` becomes `[x.data for x in arr.addressable_shards]` * `arr.device_buffers` becomes `[x.data for x in arr.addressable_shards]`
## jaxlib 0.4.22 ## jaxlib 0.4.22 (Dec 13, 2023)
## jax 0.4.21 (Dec 4 2023) ## jax 0.4.21 (Dec 4 2023)

2
jax/version.py
View File

@ -21,7 +21,7 @@ import os
import pathlib import pathlib
import subprocess import subprocess
_version = "0.4.22" _version = "0.4.23"
# The following line is overwritten by build scripts in distributions & # The following line is overwritten by build scripts in distributions &
# releases. Do not modify this manually, or jax/jaxlib build will fail. # releases. Do not modify this manually, or jax/jaxlib build will fail.
_release_version: str | None = None _release_version: str | None = None

2
setup.py
View File

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