mirror of
https://github.com/ROCm/jax.git
synced 2025-04-15 19:36:06 +00:00
Update versions post jax 0.4.18 release
This commit is contained in:
parent
cbf2cdc1bc
commit
a06beaa1a2
@ -6,9 +6,13 @@ 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.18
|
||||
# jax 0.4.19
|
||||
|
||||
# jaxlib 0.4.18
|
||||
# jaxlib 0.4.19
|
||||
|
||||
# jax 0.4.18 (Oct 6, 2023)
|
||||
|
||||
# jaxlib 0.4.18 (Oct 6, 2023)
|
||||
|
||||
* Changes
|
||||
* CUDA jaxlibs now depend on the user to install a compatible NCCL version.
|
||||
|
@ -21,7 +21,7 @@ import os
|
||||
import pathlib
|
||||
import subprocess
|
||||
|
||||
_version = "0.4.18"
|
||||
_version = "0.4.19"
|
||||
# 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
|
||||
|
2
setup.py
2
setup.py
@ -24,7 +24,7 @@ project_name = 'jax'
|
||||
|
||||
_current_jaxlib_version = '0.4.18'
|
||||
# The following should be updated with each new jaxlib release.
|
||||
_latest_jaxlib_version_on_pypi = '0.4.17'
|
||||
_latest_jaxlib_version_on_pypi = '0.4.18'
|
||||
_available_cuda11_cudnn_versions = ['86']
|
||||
_default_cuda11_cudnn_version = '86'
|
||||
_default_cuda12_cudnn_version = '89'
|
||||
|
Loading…
x
Reference in New Issue
Block a user