mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

jax 0.4.24 release take 2: update libtpu version

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2024-02-05 10:15:11 -08:00
parent 5c5e8f032a
commit 4cca5fa1da
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -28,7 +28,7 @@ _latest_jaxlib_version_on_pypi = '0.4.23'
_available_cuda11_cudnn_versions = ['86']
_default_cuda11_cudnn_version = '86'
_default_cuda12_cudnn_version = '89'
_libtpu_version = '0.1.dev20240123'
_libtpu_version = '0.1.dev20240205'
def load_version_module(pkg_path):
spec = importlib.util.spec_from_file_location(