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

Merge pull request #21765 from hawkinsp:release

Browse Source 
PiperOrigin-RevId: 641876244
This commit is contained in:
jax authors 2024-06-10 06:03:58 -07:00
commit 991797a8a9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@ -19,12 +19,12 @@ from setuptools import setup, find_packages
project_name = 'jax'
_current_jaxlib_version = '0.4.28'
_current_jaxlib_version = '0.4.29'
# The following should be updated with each new jaxlib release.
_latest_jaxlib_version_on_pypi = '0.4.28'
_default_cuda12_cudnn_version = '91'
_available_cuda12_cudnn_versions = [_default_cuda12_cudnn_version]
_libtpu_version = '0.1.dev20240508'
_libtpu_version = '0.1.dev20240609'
def load_version_module(pkg_path):
spec = importlib.util.spec_from_file_location(