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

Prepare for 0.4.26 release.

Browse Source
This commit is contained in:
Peter Hawkins 2024-04-03 14:38:58 -04:00
parent 85cb16936c
commit 61493263a9
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
CHANGELOG.md
View File

@ -50,6 +50,7 @@ Remember to align the itemized text with the first line of an item within a list
* Changes * Changes
* JAX now supports CUDA 12.1 or newer only. Support for CUDA 11.8 has been * JAX now supports CUDA 12.1 or newer only. Support for CUDA 11.8 has been
dropped. dropped.
* JAX now supports NumPy 2.0.
## jax 0.4.25 (Feb 26, 2024) ## jax 0.4.25 (Feb 26, 2024)

4
setup.py
View File

@ -22,12 +22,12 @@ from setuptools import setup, find_packages
project_name = 'jax' project_name = 'jax'
_current_jaxlib_version = '0.4.25' _current_jaxlib_version = '0.4.26'
# 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.25' _latest_jaxlib_version_on_pypi = '0.4.25'
_default_cuda12_cudnn_version = '89' _default_cuda12_cudnn_version = '89'
_available_cuda12_cudnn_versions = [_default_cuda12_cudnn_version] _available_cuda12_cudnn_versions = [_default_cuda12_cudnn_version]
_libtpu_version = '0.1.dev20240224' _libtpu_version = '0.1.dev20240403'
def load_version_module(pkg_path): def load_version_module(pkg_path):
spec = importlib.util.spec_from_file_location( spec = importlib.util.spec_from_file_location(