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 versions and changelog post 0.4.12 release

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2023-06-08 16:53:22 -07:00
parent d8571cfb6b
commit 4b80103077
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.12 ## jax 0.4.13
## jaxlib 0.4.13
## jax 0.4.12 (June 8, 2023)
* Changes * Changes
* Added {class}`scipy.spatial.transform.Rotation` and {class}`scipy.spatial.transform.Slerp` * Added {class}`scipy.spatial.transform.Rotation` and {class}`scipy.spatial.transform.Slerp`
@ -25,7 +29,7 @@ Remember to align the itemized text with the first line of an item within a list
* `jax.sharding.OpShardingSharding` has been removed since it has been 3 * `jax.sharding.OpShardingSharding` has been removed since it has been 3
months since it was deprecated. months since it was deprecated.
## jaxlib 0.4.12 ## jaxlib 0.4.12 (June 8, 2023)
* Changes * Changes
* Includes PTX/SASS for Hopper (SM version 9.0+) GPUs. Previous * Includes PTX/SASS for Hopper (SM version 9.0+) GPUs. Previous

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.4.12" __version__ = "0.4.13"
_minimum_jaxlib_version = "0.4.11" _minimum_jaxlib_version = "0.4.11"
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.4.12' _current_jaxlib_version = '0.4.12'
# 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.11' _latest_jaxlib_version_on_pypi = '0.4.12'
_available_cuda11_cudnn_versions = ['82', '86'] _available_cuda11_cudnn_versions = ['82', '86']
_default_cuda11_cudnn_version = '86' _default_cuda11_cudnn_version = '86'
_default_cuda12_cudnn_version = '88' _default_cuda12_cudnn_version = '88'