add jaxtyping v0.2.36

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

@@ -34,10 +34,12 @@ dependencies = [
     ('dm-haiku', '0.0.13', versionsuffix),
     ('RDKit', '2024.03.5'),
     ('jax-triton', '0.2.0', versionsuffix),
+    ('jaxtyping', '0.2.36', versionsuffix),
     # TODO
 ]
 
 exts_list = [
+    ('zstandard', '0.23.0'),
     ('chex', '0.1.87', {
         'checksums': ['0096d89cc8d898bb521ef4bfbf5c24549022b0e5b301f529ab57238896fe6c5d'],
     }),

easybuild/easyconfigs/j/jaxtyping/jaxtyping-0.2.36-foss-2024a-CUDA-12.6.0.eb

@@ -0,0 +1,40 @@
+# Thomas Hoffmann, EMBL Heidelberg, [email protected], 2024/12
+
+easyblock = 'PythonBundle'
+
+name = 'jaxtyping'
+version = '0.2.36'
+versionsuffix = '-CUDA-%(cudaver)s'
+
+homepage = 'https://github.com/patrick-kidger/jaxtyping'
+description = """ Type annotations and runtime type-checking for: 
+1. shape and dtype of JAX arrays;  
+   (Now also supports PyTorch, NumPy, and TensorFlow!)
+2. PyTrees.  """
+toolchain = {'name': 'foss', 'version': '2024a'}
+
+builddependencies = [
+    ('poetry', '1.8.3')
+]
+dependencies = [
+    ('Python', '3.12.3'),
+    ('CUDA', '12.6.0', '', SYSTEM),
+    ('jax', '0.4.35', versionsuffix),
+]
+
+use_pip = True
+
+exts_list = [
+    (name, version, {
+        'checksums': ['781ac44a3cf8982063d7ee48b5008ccfad7b13793bf878eb3058d5319aa08f0f'],
+    }),
+]
+
+sanity_pip_check = True
+
+sanity_check_paths = {
+    'files': [],
+    'dirs': ['lib'],
+}
+
+moduleclass = 'ai'