add tqdm dep; update jax-triton to v0.2.0

easybuild/easyconfigs/a/AlphaFold3/AlphaFold3-3.0.0-foss-2024a-CUDA-12.6.0.eb

@@ -28,11 +28,12 @@ builddependencies = [
 dependencies = [
     ('Python', '3.12.3'),
     ('CUDA', '12.6.0', '', SYSTEM),
+    ('tqdm', '4.67.0'),
     ('jax', '0.4.35', versionsuffix),
     ('dm-tree', '0.1.8'),
     ('dm-haiku', '0.0.13', versionsuffix),
     ('RDKit', '2024.03.5'),
-    ('jax-triton', '0.1.1', versionsuffix),
+    ('jax-triton', '0.2.0', versionsuffix),
     # TODO
 ]
 

easybuild/easyconfigs/j/jax-triton/jax-triton-0.2.0-foss-2024a-CUDA-12.6.0.eb

@@ -0,0 +1,36 @@
+# Thomas Hoffmann, EMBL Heidelberg, [email protected], 2024/12
+
+easyblock = 'PythonBundle'
+
+name = 'jax-triton'
+version = '0.2.0'
+versionsuffix = '-CUDA-%(cudaver)s'
+
+homepage = 'https://pypi.org/project/jax-triton'
+description = """The jax-triton repository contains integrations between JAX and Triton."""
+
+toolchain = {'name': 'foss', 'version': '2024a'}
+
+dependencies = [
+    ('Python', '3.12.3'),
+    ('CUDA', '12.6.0', '', SYSTEM),
+    ('jax', '0.4.35', versionsuffix),
+    ('Triton', '3.1.0', versionsuffix),
+]
+
+use_pip = True
+
+exts_list = [
+    ('jax_triton', version, {
+        'checksums': ['ed564a5ffb9e404557dc8d296e7eb30e501da5a4d7b03408acd0837c1c618c21'],
+    }),
+]
+
+sanity_pip_check = True
+
+sanity_check_paths = {
+    'files': [],
+    'dirs': ['lib'],
+}
+
+moduleclass = 'ai'