1
0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00

Finish jax and jaxlib 0.4.27 release

PiperOrigin-RevId: 631486157
This commit is contained in:
Yash Katariya 2024-05-07 11:13:21 -07:00 committed by jax authors
parent 3dd8af93b4
commit 5031a1ddc4
3 changed files with 8 additions and 4 deletions

@ -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.
-->
## jax 0.4.27
## jax 0.4.28
## jaxlib 0.4.28
## jax 0.4.27 (May 7, 2024)
* New Functionality
* Added {func}`jax.numpy.unstack` and {func}`jax.numpy.cumulative_sum`,
@ -78,7 +82,7 @@ Remember to align the itemized text with the first line of an item within a list
dtype as the input array. This may result in some increased memory usage.
The default value is set to `copy=False` to preserve backwards compatability.
## jaxlib 0.4.27
## jaxlib 0.4.27 (May 7, 2024)
## jax 0.4.26 (April 3, 2024)

@ -21,7 +21,7 @@ import os
import pathlib
import subprocess
_version = "0.4.27"
_version = "0.4.28"
# The following line is overwritten by build scripts in distributions &
# releases. Do not modify this manually, or jax/jaxlib build will fail.
_release_version: str | None = None

@ -24,7 +24,7 @@ project_name = 'jax'
_current_jaxlib_version = '0.4.27'
# The following should be updated with each new jaxlib release.
_latest_jaxlib_version_on_pypi = '0.4.26'
_latest_jaxlib_version_on_pypi = '0.4.27'
_default_cuda12_cudnn_version = '89'
_available_cuda12_cudnn_versions = [_default_cuda12_cudnn_version]
_libtpu_version = '0.1.dev20240507'