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

Lower the minimum jaxlib version to 0.4.31, since 0.4.32 was yanked from pypi.

Browse Source 
PiperOrigin-RevId: 673992844
This commit is contained in:
Peter Hawkins 2024-09-12 13:45:33 -07:00 committed by jax authors
parent 39c39acd19
commit 46a7be6d7a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/version.py
View File

@ -133,7 +133,7 @@ def _get_cmdclass(pkg_source_path):
__version__ = _get_version_string()
_minimum_jaxlib_version = "0.4.32"
_minimum_jaxlib_version = "0.4.31"
def _version_as_tuple(version_str):
return tuple(int(i) for i in version_str.split(".") if i.isdigit())