From 6868be30b716b659372a62999b791c180c569aa5 Mon Sep 17 00:00:00 2001 From: Yash Katariya Date: Thu, 17 Aug 2023 19:02:24 -0700 Subject: [PATCH] Create a jax nightly wheel and upload to GCS. PiperOrigin-RevId: 557998568 --- jax/version.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/jax/version.py b/jax/version.py index 79d908205..1e10e3397 100644 --- a/jax/version.py +++ b/jax/version.py @@ -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):