mirror of
https://github.com/ROCm/jax.git
synced 2025-04-15 19:36:06 +00:00
Update jaxlib version etc.
This commit is contained in:
parent
23099f6007
commit
f06bb9a7f4
@ -15,6 +15,8 @@ PLEASE REMEMBER TO CHANGE THE '..master' WITH AN ACTUAL TAG in GITHUB LINK.
|
||||
* Bug fixes:
|
||||
* Breaking changes:
|
||||
|
||||
## jaxlib 0.1.63 (Unreleased)
|
||||
|
||||
## jax 0.2.10 (March 5 2021)
|
||||
|
||||
* [GitHub commits](https://github.com/google/jax/compare/jax-v0.2.9...jax-v0.2.10).
|
||||
@ -46,7 +48,7 @@ PLEASE REMEMBER TO CHANGE THE '..master' WITH AN ACTUAL TAG in GITHUB LINK.
|
||||
{func}`jax.numpy.repeat`, {func}`jax.numpy.reshape`.
|
||||
In general, {mod}`jax.numpy` functions should be used with scalars or array arguments.
|
||||
|
||||
## jaxlib 0.1.62 (Unreleased)
|
||||
## jaxlib 0.1.62 (March 9 2021)
|
||||
|
||||
* New features:
|
||||
* jaxlib wheels are now built to require AVX instructions on x86-64 machines
|
||||
@ -571,7 +573,7 @@ PLEASE REMEMBER TO CHANGE THE '..master' WITH AN ACTUAL TAG in GITHUB LINK.
|
||||
> > * Forward-mode automatic differentiation (`jvp`) of while loop
|
||||
> ({jax-issue}`#1980`)
|
||||
> * New NumPy and SciPy functions:
|
||||
>
|
||||
>
|
||||
> * {py:func}`jax.numpy.fft.fft2`
|
||||
> * {py:func}`jax.numpy.fft.ifft2`
|
||||
> * {py:func}`jax.numpy.fft.rfft`
|
||||
@ -592,5 +594,3 @@ PLEASE REMEMBER TO CHANGE THE '..master' WITH AN ACTUAL TAG in GITHUB LINK.
|
||||
|
||||
* With the Python 3 upgrade, JAX no longer depends on `fastcache`, which should
|
||||
help with installation.
|
||||
|
||||
|
||||
|
@ -422,7 +422,7 @@ Next, run
|
||||
|
||||
```bash
|
||||
pip install --upgrade pip
|
||||
pip install --upgrade jax jaxlib==0.1.61+cuda110 -f https://storage.googleapis.com/jax-releases/jax_releases.html
|
||||
pip install --upgrade jax jaxlib==0.1.62+cuda110 -f https://storage.googleapis.com/jax-releases/jax_releases.html
|
||||
```
|
||||
|
||||
The jaxlib version must correspond to the version of the existing CUDA
|
||||
|
@ -14,4 +14,4 @@
|
||||
|
||||
# This should be increased after releasing the current version (i.e. this
|
||||
# is always the next version to be released).
|
||||
__version__ = "0.1.62"
|
||||
__version__ = "0.1.63"
|
||||
|
Loading…
x
Reference in New Issue
Block a user