diff --git a/Golden_Repo/j/JAX/JAX-0.2.9-gcccoremkl-9.3.0-2020.2.254-Python-3.8.5.eb b/Golden_Repo/j/JAX/JAX-0.2.11-gcccoremkl-9.3.0-2020.2.254-Python-3.8.5.eb similarity index 74% rename from Golden_Repo/j/JAX/JAX-0.2.9-gcccoremkl-9.3.0-2020.2.254-Python-3.8.5.eb rename to Golden_Repo/j/JAX/JAX-0.2.11-gcccoremkl-9.3.0-2020.2.254-Python-3.8.5.eb index 33eaf5774bbe35d4a85032d78c56d9264a9dc62e..935212ea5c636e12cb2c221d935084f52191a942 100644 --- a/Golden_Repo/j/JAX/JAX-0.2.9-gcccoremkl-9.3.0-2020.2.254-Python-3.8.5.eb +++ b/Golden_Repo/j/JAX/JAX-0.2.11-gcccoremkl-9.3.0-2020.2.254-Python-3.8.5.eb @@ -2,7 +2,7 @@ import os easyblock = 'PythonPackage' name = 'JAX' -version = '0.2.9' +version = '0.2.11' versionsuffix = '-Python-%(pyver)s' local_cudaver = '11.0' @@ -22,9 +22,10 @@ dependencies = [ ('libjpeg-turbo', '2.0.5'), ] -source_urls = [PYPI_SOURCE] -sources = [SOURCELOWER_TAR_GZ] -checksums = ['c563c97fae7b85b0f5622522f9fbdf61e04bf0d1e50888a644bdc966b5c1f274'] +github_account = 'google' +source_urls = [GITHUB_LOWER_SOURCE] +sources = ["%(namelower)s-v%(version)s.tar.gz"] +checksums = ['82505dca6c3171e0e5b6ba4dfe69e0322e27dc66e6bdd4a78e6119e3148ae413'] download_dep_fail = True use_pip = True @@ -44,15 +45,15 @@ exts_list = [ 'unpack_sources': False, 'checksums': ['9e9ef47fa92625c4721036e7c4124182668dc6021d9e7c73704edd395648deb9'], }), - ('absl-py', '0.10.0', { + ('absl-py', '0.12.0', { 'modulename': 'absl', - 'checksums': ['b20f504a7871a580be5268a18fbad48af4203df5d33dbc9272426cb806245a45'], + 'checksums': ['b44f68984a5ceb2607d135a615999b93924c771238a63920d17d3387b0d229d5'], }), - ('jaxlib', '0.1.61', { - 'source_urls': 'https://storage.googleapis.com/jax-releases/cuda110/', - 'source_tmpl': 'jaxlib-0.1.61+cuda110-cp38-none-manylinux2010_x86_64.whl', + ('jaxlib', '0.1.64', { + 'source_tmpl': 'jaxlib-0.1.64+cuda110-cp38-none-manylinux2010_x86_64.whl', + 'source_urls': ['https://storage.googleapis.com/jax-releases/cuda110/'], 'unpack_sources': False, - 'checksums': ['64ae2637367fae1474b443705c61a422acb646c5965eb98d466b4c0e38eb86b7'], + 'checksums': ['36f20d9d54482da7239f38e59d5301a15a64a56fa35b6c8ba31f7871504dba2a'], }), ('opt_einsum', '3.3.0', { 'source_tmpl': 'opt_einsum-%(version)s-py3-none-any.whl', @@ -61,7 +62,7 @@ exts_list = [ }), ] -# This should be modextravars, but life is unfair +# This should be modextravars, but life is unfair # Jaxlib searches in hardcoded paths for CUDA, or in those variables modluafooter = """ local cuda_root = os.getenv("EBROOTCUDA") or ""