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

Create a jax nightly wheel and upload to GCS.

Browse Source 
PiperOrigin-RevId: 557998568
This commit is contained in:
Yash Katariya 2023-08-17 19:02:24 -07:00 committed by jax authors
parent 6d1b5d90ae
commit 6868be30b7
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
jax/version.py
View File

@ -15,7 +15,15 @@
# This file is included as part of both jax and jaxlib. It is also
# eval()-ed by setup.py, so it should not have any dependencies.
import os
import datetime
__version__ = "0.4.15"
if os.environ.get('JAX_NIGHTLY'):
datestring = datetime.datetime.now().strftime('%Y%m%d')
__version__ = f'{__version__}.dev{datestring}'
_minimum_jaxlib_version = "0.4.14"
def _version_as_tuple(version_str):