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 CHANGELOG.md and setup.py for 0.3.14 release.

Browse Source
This commit is contained in:
Peter Hawkins 2022-06-27 09:21:54 -04:00
parent 02603606e7
commit 1e29b7b762
2 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
CHANGELOG.md
View File

@ -11,18 +11,17 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
## jax 0.3.15 (Unreleased) ## jax 0.3.15 (Unreleased)
* [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.14...main). * [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.14...main).
* Changes * Changes
* {func}`jax.numpy.roots` is now better behaved when `strip_zeros=False` when
coefficients have leading zeros ({jax-issue}`#11215`).
## jaxlib 0.3.15 (Unreleased) ## jaxlib 0.3.15 (Unreleased)
## jax 0.3.14 (June 21, 2022) ## jax 0.3.14 (June 27, 2022)
* [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.13...jax-v0.3.14). * [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.13...jax-v0.3.14).
* Breaking changes * Breaking changes
* {func}`jax.experimental.compilation_cache.initialize_cache` does not support * {func}`jax.experimental.compilation_cache.initialize_cache` does not support
`max_cache_size_ bytes` anymore and will not get that as an input. `max_cache_size_ bytes` anymore and will not get that as an input.
* `JAX_PLATFORMS` now raises an exception when platform initialization fails. * `JAX_PLATFORMS` now raises an exception when platform initialization fails.
* Changes * Changes
* Fixed compatibility problems with NumPy 1.23.
* {func}`jax.numpy.linalg.slogdet` now accepts an optional `method` argument * {func}`jax.numpy.linalg.slogdet` now accepts an optional `method` argument
that allows selection between an LU-decomposition based implementation and that allows selection between an LU-decomposition based implementation and
an implementation based on QR decomposition. an implementation based on QR decomposition.
@ -65,11 +64,15 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
* {func}`jax.experimental.compilation_cache.initialize_cache` now supports gcs * {func}`jax.experimental.compilation_cache.initialize_cache` now supports gcs
bucket path as input. bucket path as input.
* Added {func}`jax.scipy.stats.gennorm`. * Added {func}`jax.scipy.stats.gennorm`.
* {func}`jax.numpy.roots` is now better behaved when `strip_zeros=False` when
coefficients have leading zeros ({jax-issue}`#11215`).
## jaxlib 0.3.14 (June 21, 2022) ## jaxlib 0.3.14 (June 27, 2022)
* [GitHub commits](https://github.com/google/jax/compare/jaxlib-v0.3.10...main). * [GitHub commits](https://github.com/google/jax/compare/jaxlib-v0.3.10...jaxlib-v0.3.14).
* x86-64 Mac wheels now require Mac OS 10.14 (Mojave) or newer. Mac OS 10.14 * x86-64 Mac wheels now require Mac OS 10.14 (Mojave) or newer. Mac OS 10.14
was released in 2018, so this should not be a very onerous requirement. was released in 2018, so this should not be a very onerous requirement.
* The bundled version of NCCL was updated to 2.12.12, fixing some deadlocks.
* The Python flatbuffers package is no longer a dependency of jaxlib.
## jax 0.3.13 (May 16, 2022) ## jax 0.3.13 (May 16, 2022)
* [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.12...jax-v0.3.13). * [GitHub commits](https://github.com/google/jax/compare/jax-v0.3.12...jax-v0.3.13).

2
setup.py
View File

@ -21,7 +21,7 @@ _available_cuda_versions = ['11']
_default_cuda_version = '11' _default_cuda_version = '11'
_available_cudnn_versions = ['82', '805'] _available_cudnn_versions = ['82', '805']
_default_cudnn_version = '82' _default_cudnn_version = '82'
_libtpu_version = '0.1.dev20220621' _libtpu_version = '0.1.dev20220627'
_dct = {} _dct = {}
with open('jax/version.py') as f: with open('jax/version.py') as f: