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

Update files after new jaxlib release 0.1.71

Browse Source
This commit is contained in:
yashkatariya 2021-09-01 10:43:20 -07:00
parent b10fb54eec
commit be824a792e
3 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
CHANGELOG.md
View File

@ -16,6 +16,11 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
* `jnp.unique` and other set-like operations now require array-like inputs
({jax-issue}`#7662`)
## jaxlib 0.1.71 (Sep 1, 2021)
* Breaking changes:
* Support for CUDA 11.0 and CUDA 10.1 has been dropped. Jaxlib now supports
CUDA 10.2 and CUDA 11.1+.
## jax 0.2.19 (Aug 12, 2021)
* [GitHub
commits](https://github.com/google/jax/compare/jax-v0.2.18...jax-v0.2.19).

2
jaxlib/version.py
View File

@ -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.71"
__version__ = "0.1.72"

4
setup.py
View File

@ -15,8 +15,8 @@
from setuptools import setup, find_packages
# The following should be updated with each new jaxlib release.
_current_jaxlib_version = '0.1.70'
_available_cuda_versions = ['101', '102', '110', '111']
_current_jaxlib_version = '0.1.71'
_available_cuda_versions = ['102', '111']
_dct = {}
with open('jax/version.py') as f: