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

Update version.py, setup.py and CHANGELOG post jax 0.3.22 release

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2022-10-11 21:37:53 +00:00
parent 1fb886ce43
commit a6e0e77624
3 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
CHANGELOG.md
View File

@ -8,8 +8,11 @@ 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. PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
--> -->
## jax 0.3.22 ## jax 0.3.23
* [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.21...main).
## jaxlib 0.3.23
## jax 0.3.22 (Oct 11, 2022)
* Changes * Changes
* Add `JAX_PLATFORMS=tpu,cpu` as default setting in TPU initialization, * Add `JAX_PLATFORMS=tpu,cpu` as default setting in TPU initialization,
so JAX will raise an error if TPU cannot be initialized instead of falling so JAX will raise an error if TPU cannot be initialized instead of falling
@ -20,8 +23,7 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
* Several test utilities deprecated in JAX v0.3.8 are now removed from * Several test utilities deprecated in JAX v0.3.8 are now removed from
{mod}`jax.test_util`. {mod}`jax.test_util`.
## jaxlib 0.3.22 ## jaxlib 0.3.22 (Oct 11, 2022)
* [GitHub commits](https://github.com/google/jax/compare/jaxlib-v0.3.20...main).
## jax 0.3.21 (Sep 30, 2022) ## jax 0.3.21 (Sep 30, 2022)
* [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.20...jax-v0.3.21). * [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.20...jax-v0.3.21).

2
jax/version.py
View File

@ -15,7 +15,7 @@
# This file is included as part of both jax and jaxlib. It is also # 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. # eval()-ed by setup.py, so it should not have any dependencies.
__version__ = "0.3.22" __version__ = "0.3.23"
_minimum_jaxlib_version = "0.3.15" _minimum_jaxlib_version = "0.3.15"
def _version_as_tuple(version_str): def _version_as_tuple(version_str):

2
setup.py
View File

@ -21,7 +21,7 @@ from setuptools import setup, find_packages
_current_jaxlib_version = '0.3.22' _current_jaxlib_version = '0.3.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.3.20' _latest_jaxlib_version_on_pypi = '0.3.22'
_available_cuda_versions = ['11'] _available_cuda_versions = ['11']
_default_cuda_version = '11' _default_cuda_version = '11'
_available_cudnn_versions = ['82', '805'] _available_cudnn_versions = ['82', '805']