mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update libtpu version for jaxlib 0.4.2 release (again)

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2023-01-25 01:34:16 +00:00
parent 09794bee5f
commit c4ad27c363
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

@ -26,7 +26,7 @@ _available_cuda_versions = ['11']
_default_cuda_version = '11'
_available_cudnn_versions = ['82', '86']
_default_cudnn_version = '86'
_libtpu_version = '0.1.dev20230120'
_libtpu_version = '0.1.dev20230124'
_dct = {}
with open('jax/version.py', encoding='utf-8') as f: