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 jax 0.4.32 release

Browse Source 
PiperOrigin-RevId: 673105544
This commit is contained in:
Yash Katariya 2024-09-10 14:25:40 -07:00 committed by jax authors
parent 4957ab9a5e
commit 5e4250e64b
2 changed files with 3 additions and 3 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.31"
_minimum_jaxlib_version = "0.4.32"
def _version_as_tuple(version_str):
return tuple(int(i) for i in version_str.split(".") if i.isdigit())

4
setup.py
View File

@ -19,10 +19,10 @@ from setuptools import setup, find_packages
project_name = 'jax'
_current_jaxlib_version = '0.4.31'
_current_jaxlib_version = '0.4.32'
# The following should be updated after each new jaxlib release.
_latest_jaxlib_version_on_pypi = '0.4.31'
_libtpu_version = '0.1.dev20240729'
_libtpu_version = '0.1.dev20240910'
def load_version_module(pkg_path):
spec = importlib.util.spec_from_file_location(