{data}[fosscuda/2019b] JAX v0.1.77 w/ Python 3.7.4

easybuild/easyconfigs/b/Bazel/Bazel-2.0.0-GCCcore-8.3.0.eb

@@ -15,6 +15,12 @@ checksums = [
     '99928d0902beeaf962a8ad14db8432f8e5114645e3caf64c7ee2fa136c31609f',  # Bazel-0.29.1_fix-gold-flag.patch
 ]
 
+# Bazel doesn't pass LD_LIBRARY_PATH etc. through, even to its own "process-wrapper".
+# So we'll statically link bazel, to avoid errors like:
+#     "process-wrapper: /lib64/libstdc++.so.6: version `GLIBCXX_3.4.21' not found"
+# See https://github.com/bazelbuild/bazel/issues/4137
+static = True
+
 builddependencies = [
     ('binutils', '2.32'),
     ('Python', '3.7.4'),

easybuild/easyconfigs/j/JAX/JAX-0.1.77-fosscuda-2019b-Python-3.7.4.eb

@@ -0,0 +1,23 @@
+# This easyconfig was created by the BEAR Software team at the University of Birmingham.
+name = 'JAX'
+version = '0.1.77'
+versionsuffix = '-Python-%(pyver)s'
+
+homepage = "https://github.com/google/jax"
+description = """JAX is Autograd and XLA, brought together for high-performance machine learning research."""
+
+toolchain = {'name': 'fosscuda', 'version': '2019b'}
+
+builddependencies = [('Bazel', '2.0.0')]
+
+dependencies = [
+    ('Python', '3.7.4'),
+    ('SciPy-bundle', '2019.10', versionsuffix),
+    ('cuDNN', '7.6.4.38'),
+]
+
+source_urls = ['https://github.com/google/jax/archive']
+sources = ['jax-v%(version)s.tar.gz']
+checksums = ['18fa3bcebf4b1a526349db5a76a508784ec5d05324c192c40f240ccbf9480e26']
+
+moduleclass = 'data'