Skip to content

{lib}[foss/2024a] dm-haiku v0.0.13 w/ CUDA 12.6.0#22028

Open
ThomasHoffmann77 wants to merge 20 commits intoeasybuilders:developfrom
ThomasHoffmann77:20241213170429_new_pr_dm-haiku0013
Open

{lib}[foss/2024a] dm-haiku v0.0.13 w/ CUDA 12.6.0#22028
ThomasHoffmann77 wants to merge 20 commits intoeasybuilders:developfrom
ThomasHoffmann77:20241213170429_new_pr_dm-haiku0013

Conversation

@ThomasHoffmann77
Copy link
Contributor

@ThomasHoffmann77 ThomasHoffmann77 commented Dec 13, 2024

…store-0.1.69-foss-2024a.eb and patches: tensorstore-0.1.69_disable_third_party_downloads.patch
@github-actions
Copy link

github-actions bot commented Dec 13, 2024

Updated software dm-haiku-0.0.13-foss-2024a-CUDA-12.6.0.eb

Diff against dm-haiku-0.0.13-foss-2023a.eb

easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.13-foss-2023a.eb

diff --git a/easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.13-foss-2023a.eb b/easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.13-foss-2024a-CUDA-12.6.0.eb
index 9e73940419..ff19de7c00 100644
--- a/easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.13-foss-2023a.eb
+++ b/easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.13-foss-2024a-CUDA-12.6.0.eb
@@ -1,29 +1,51 @@
+# update 0.0.12: Thomas Hoffmann (EMBL)
 easyblock = 'PythonBundle'
 
 name = 'dm-haiku'
 version = '0.0.13'
+versionsuffix = '-CUDA-%(cudaver)s'
 
 homepage = 'https://github.com/deepmind/dm-haiku'
 description = """Haiku is a simple neural network library for JAX developed by some of the authors of Sonnet, a neural
 network library for TensorFlow."""
 
-toolchain = {'name': 'foss', 'version': '2023a'}
+toolchain = {'name': 'foss', 'version': '2024a'}
+
+builddependencies = [
+    ('poetry', '1.8.3')
+]
 
 dependencies = [
-    ('Python', '3.11.3'),
-    ('SciPy-bundle', '2023.07'),
-    ('jax', '0.4.25'),  # required by jmp, also provides absl-py
+    ('Python', '3.12.3'),
+    ('SciPy-bundle', '2024.05'),
+    ('jax', '0.4.35', versionsuffix),  # required by jmp, also provides absl-py
+    ('PyYAML', '6.0.2'),
+    ('CUDA', '12.6.0', '', SYSTEM),
+    ('tensorstore', '0.1.72'),
+    ('protobuf-python', '5.28.0'),
+    ('Optax', '0.2.4', versionsuffix),
 ]
 
 exts_list = [
     ('jmp', '0.0.4', {
-        'source_tmpl': SOURCE_PY3_WHL,
-        'checksums': ['6aa7adbddf2bd574b28c7faf6e81a735eb11f53386447896909c6968dc36807d'],
+        'checksums': ['5dfeb0fd7c7a9f72a70fff0aab9d0cbfae32a809c02f4037ff3485ceb33e1730'],
+    }),
+    ('flax', '0.10.2', {
+        'checksums': ['6f831350026ad48182ba6588bb4dd72dc1084985d9aca923254cb3e4c78d75f3'],
+    }),
+    ('nest_asyncio', '1.6.0', {
+        'checksums': ['6f172d5449aca15afd6c646851f4e31e02c598d553a667e38cafa997cfec55fe'],
+    }),
+    ('humanize', '4.11.0', {
+        'checksums': ['e66f36020a2d5a974c504bd2555cf770621dbdbb6d82f94a6857c0b1ea2608be'],
+    }),
+    ('orbax_checkpoint', '0.10.2', {
+        'modulename': 'orbax.checkpoint',
+        'checksums': ['e575ebe1f94e5cb6353ab8c9df81de0ca7cddc118645c3bfc17b8344f19d42f1'],
     }),
     ('dm_haiku', version, {
         'modulename': 'haiku',
-        'source_tmpl': SOURCE_PY3_WHL,
-        'checksums': ['ee9562c68a059f146ad07f555ca591cb8c11ef751afecc38353863562bd23f43'],
+        'checksums': ['029bb91b5b1edb0d3fe23304d3bf12a545ea6e485041f7f5d8c8d85ebcf6e17d'],
     }),
 ]
 
Diff against dm-haiku-0.0.12-foss-2023a-CUDA-12.1.1.eb

easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.12-foss-2023a-CUDA-12.1.1.eb

diff --git a/easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.12-foss-2023a-CUDA-12.1.1.eb b/easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.13-foss-2024a-CUDA-12.6.0.eb
index 76672f6fb7..ff19de7c00 100644
--- a/easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.12-foss-2023a-CUDA-12.1.1.eb
+++ b/easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.13-foss-2024a-CUDA-12.6.0.eb
@@ -2,44 +2,50 @@
 easyblock = 'PythonBundle'
 
 name = 'dm-haiku'
-version = '0.0.12'
+version = '0.0.13'
 versionsuffix = '-CUDA-%(cudaver)s'
 
 homepage = 'https://github.com/deepmind/dm-haiku'
 description = """Haiku is a simple neural network library for JAX developed by some of the authors of Sonnet, a neural
 network library for TensorFlow."""
 
-toolchain = {'name': 'foss', 'version': '2023a'}
+toolchain = {'name': 'foss', 'version': '2024a'}
+
+builddependencies = [
+    ('poetry', '1.8.3')
+]
 
 dependencies = [
-    ('Python', '3.11.3'),
-    ('SciPy-bundle', '2023.07'),
-    ('jax', '0.4.25', versionsuffix),  # required by jmp, also provides absl-py
-    ('PyYAML', '6.0'),
-    ('CUDA', '12.1.1', '', SYSTEM),
-    ('tensorstore', '0.1.65'),
-    ('protobuf-python', '4.24.0'),
-    ('Optax', '0.2.2', versionsuffix),
+    ('Python', '3.12.3'),
+    ('SciPy-bundle', '2024.05'),
+    ('jax', '0.4.35', versionsuffix),  # required by jmp, also provides absl-py
+    ('PyYAML', '6.0.2'),
+    ('CUDA', '12.6.0', '', SYSTEM),
+    ('tensorstore', '0.1.72'),
+    ('protobuf-python', '5.28.0'),
+    ('Optax', '0.2.4', versionsuffix),
 ]
 
 exts_list = [
     ('jmp', '0.0.4', {
         'checksums': ['5dfeb0fd7c7a9f72a70fff0aab9d0cbfae32a809c02f4037ff3485ceb33e1730'],
     }),
-    ('flax', '0.8.4', {
-        'checksums': ['968683f850198e1aa5eb2d9d1e20bead880ef7423c14f042db9d60848cb1c90b'],
+    ('flax', '0.10.2', {
+        'checksums': ['6f831350026ad48182ba6588bb4dd72dc1084985d9aca923254cb3e4c78d75f3'],
     }),
     ('nest_asyncio', '1.6.0', {
         'checksums': ['6f172d5449aca15afd6c646851f4e31e02c598d553a667e38cafa997cfec55fe'],
     }),
-    ('orbax_checkpoint', '0.5.18', {
+    ('humanize', '4.11.0', {
+        'checksums': ['e66f36020a2d5a974c504bd2555cf770621dbdbb6d82f94a6857c0b1ea2608be'],
+    }),
+    ('orbax_checkpoint', '0.10.2', {
         'modulename': 'orbax.checkpoint',
-        'preinstallopts': """sed -i 's/jax >= 0.4.25/&\\*/g' pyproject.toml &&""",
-        'checksums': ['29f5d311b412760bd6a2fecab3bdbf75407bc00dc6d0457d19478258ecc8fa6d'],
+        'checksums': ['e575ebe1f94e5cb6353ab8c9df81de0ca7cddc118645c3bfc17b8344f19d42f1'],
     }),
-    (name, version, {
+    ('dm_haiku', version, {
         'modulename': 'haiku',
-        'checksums': ['ba0b3acf71433156737fe342c486da11727e5e6c9e054245f4f9b8f0b53eb608'],
+        'checksums': ['029bb91b5b1edb0d3fe23304d3bf12a545ea6e485041f7f5d8c8d85ebcf6e17d'],
     }),
 ]
 
Diff against dm-haiku-0.0.9-foss-2022a-CUDA-11.7.0.eb

easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.9-foss-2022a-CUDA-11.7.0.eb

diff --git a/easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.9-foss-2022a-CUDA-11.7.0.eb b/easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.13-foss-2024a-CUDA-12.6.0.eb
index eb44afa10f..ff19de7c00 100644
--- a/easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.9-foss-2022a-CUDA-11.7.0.eb
+++ b/easybuild/easyconfigs/d/dm-haiku/dm-haiku-0.0.13-foss-2024a-CUDA-12.6.0.eb
@@ -1,29 +1,51 @@
+# update 0.0.12: Thomas Hoffmann (EMBL)
 easyblock = 'PythonBundle'
 
 name = 'dm-haiku'
-version = '0.0.9'
+version = '0.0.13'
 versionsuffix = '-CUDA-%(cudaver)s'
 
 homepage = 'https://github.com/deepmind/dm-haiku'
 description = """Haiku is a simple neural network library for JAX developed by some of the authors of Sonnet, a neural
 network library for TensorFlow."""
 
-toolchain = {'name': 'foss', 'version': '2022a'}
+toolchain = {'name': 'foss', 'version': '2024a'}
+
+builddependencies = [
+    ('poetry', '1.8.3')
+]
 
 dependencies = [
-    ('CUDA', '11.7.0', '', SYSTEM),
-    ('Python', '3.10.4'),
-    ('SciPy-bundle', '2022.05'),
-    ('jax', '0.3.25', versionsuffix),  # required by jmp, also provides absl-py
+    ('Python', '3.12.3'),
+    ('SciPy-bundle', '2024.05'),
+    ('jax', '0.4.35', versionsuffix),  # required by jmp, also provides absl-py
+    ('PyYAML', '6.0.2'),
+    ('CUDA', '12.6.0', '', SYSTEM),
+    ('tensorstore', '0.1.72'),
+    ('protobuf-python', '5.28.0'),
+    ('Optax', '0.2.4', versionsuffix),
 ]
 
 exts_list = [
     ('jmp', '0.0.4', {
         'checksums': ['5dfeb0fd7c7a9f72a70fff0aab9d0cbfae32a809c02f4037ff3485ceb33e1730'],
     }),
-    (name, version, {
+    ('flax', '0.10.2', {
+        'checksums': ['6f831350026ad48182ba6588bb4dd72dc1084985d9aca923254cb3e4c78d75f3'],
+    }),
+    ('nest_asyncio', '1.6.0', {
+        'checksums': ['6f172d5449aca15afd6c646851f4e31e02c598d553a667e38cafa997cfec55fe'],
+    }),
+    ('humanize', '4.11.0', {
+        'checksums': ['e66f36020a2d5a974c504bd2555cf770621dbdbb6d82f94a6857c0b1ea2608be'],
+    }),
+    ('orbax_checkpoint', '0.10.2', {
+        'modulename': 'orbax.checkpoint',
+        'checksums': ['e575ebe1f94e5cb6353ab8c9df81de0ca7cddc118645c3bfc17b8344f19d42f1'],
+    }),
+    ('dm_haiku', version, {
         'modulename': 'haiku',
-        'checksums': ['97752b32cdbe5a3e2d1c60ea884d33eb4b36e7d410000f0d61b571417c925435'],
+        'checksums': ['029bb91b5b1edb0d3fe23304d3bf12a545ea6e485041f7f5d8c8d85ebcf6e17d'],
     }),
 ]
 

Updated software Optax-0.2.4-gfbf-2024a-CUDA-12.6.0.eb

Diff against Optax-0.2.2-gfbf-2023a-CUDA-12.1.1.eb

easybuild/easyconfigs/o/Optax/Optax-0.2.2-gfbf-2023a-CUDA-12.1.1.eb

diff --git a/easybuild/easyconfigs/o/Optax/Optax-0.2.2-gfbf-2023a-CUDA-12.1.1.eb b/easybuild/easyconfigs/o/Optax/Optax-0.2.4-gfbf-2024a-CUDA-12.6.0.eb
index 45b587bc0d..d8c53f9d2b 100644
--- a/easybuild/easyconfigs/o/Optax/Optax-0.2.2-gfbf-2023a-CUDA-12.1.1.eb
+++ b/easybuild/easyconfigs/o/Optax/Optax-0.2.4-gfbf-2024a-CUDA-12.6.0.eb
@@ -1,30 +1,30 @@
 easyblock = 'PythonBundle'
 
 name = 'Optax'
-version = '0.2.2'
+version = '0.2.4'
 versionsuffix = '-CUDA-%(cudaver)s'
 
 homepage = 'https://github.com/deepmind/optax'
 description = """Optax is a gradient processing and optimization library for JAX."""
 
-toolchain = {'name': 'gfbf', 'version': '2023a'}
+toolchain = {'name': 'gfbf', 'version': '2024a'}
 
 dependencies = [
-    ('CUDA', '12.1.1', '', SYSTEM),
-    ('Python', '3.11.3'),
-    ('SciPy-bundle', '2023.07'),
-    ('jax', '0.4.25', versionsuffix),
+    ('CUDA', '12.6.0', '', SYSTEM),
+    ('Python', '3.12.3'),
+    ('SciPy-bundle', '2024.05'),
+    ('jax', '0.4.35', versionsuffix),
 ]
 
 exts_list = [
-    ('toolz', '0.12.1', {
-        'checksums': ['ecca342664893f177a13dac0e6b41cbd8ac25a358e5f215316d43e2100224f4d'],
+    ('chex', '0.1.87', {
+        'checksums': ['0096d89cc8d898bb521ef4bfbf5c24549022b0e5b301f529ab57238896fe6c5d'],
     }),
-    ('chex', '0.1.86', {
-        'checksums': ['e8b0f96330eba4144659e1617c0f7a57b161e8cbb021e55c6d5056c7378091d1'],
+    ('toolz', '1.0.0', {
+        'checksums': ['2c86e3d9a04798ac556793bced838816296a2f085017664e4995cb40a1047a02'],
     }),
     ('optax', version, {
-        'checksums': ['f09bf790ef4b09fb9c35f79a07594c6196a719919985f542dc84b0bf97812e0e'],
+        'checksums': ['4e05d3d5307e6dde4c319187ae36e6cd3a0c035d4ed25e9e992449a304f47336'],
     }),
 ]
 
Diff against Optax-0.2.2-gfbf-2023a.eb

easybuild/easyconfigs/o/Optax/Optax-0.2.2-gfbf-2023a.eb

diff --git a/easybuild/easyconfigs/o/Optax/Optax-0.2.2-gfbf-2023a.eb b/easybuild/easyconfigs/o/Optax/Optax-0.2.4-gfbf-2024a-CUDA-12.6.0.eb
index 007b56ab55..d8c53f9d2b 100644
--- a/easybuild/easyconfigs/o/Optax/Optax-0.2.2-gfbf-2023a.eb
+++ b/easybuild/easyconfigs/o/Optax/Optax-0.2.4-gfbf-2024a-CUDA-12.6.0.eb
@@ -1,28 +1,30 @@
 easyblock = 'PythonBundle'
 
 name = 'Optax'
-version = '0.2.2'
+version = '0.2.4'
+versionsuffix = '-CUDA-%(cudaver)s'
 
 homepage = 'https://github.com/deepmind/optax'
 description = """Optax is a gradient processing and optimization library for JAX."""
 
-toolchain = {'name': 'gfbf', 'version': '2023a'}
+toolchain = {'name': 'gfbf', 'version': '2024a'}
 
 dependencies = [
-    ('Python', '3.11.3'),
-    ('SciPy-bundle', '2023.07'),
-    ('jax', '0.4.25'),
+    ('CUDA', '12.6.0', '', SYSTEM),
+    ('Python', '3.12.3'),
+    ('SciPy-bundle', '2024.05'),
+    ('jax', '0.4.35', versionsuffix),
 ]
 
 exts_list = [
-    ('toolz', '0.12.1', {
-        'checksums': ['ecca342664893f177a13dac0e6b41cbd8ac25a358e5f215316d43e2100224f4d'],
+    ('chex', '0.1.87', {
+        'checksums': ['0096d89cc8d898bb521ef4bfbf5c24549022b0e5b301f529ab57238896fe6c5d'],
     }),
-    ('chex', '0.1.86', {
-        'checksums': ['e8b0f96330eba4144659e1617c0f7a57b161e8cbb021e55c6d5056c7378091d1'],
+    ('toolz', '1.0.0', {
+        'checksums': ['2c86e3d9a04798ac556793bced838816296a2f085017664e4995cb40a1047a02'],
     }),
     ('optax', version, {
-        'checksums': ['f09bf790ef4b09fb9c35f79a07594c6196a719919985f542dc84b0bf97812e0e'],
+        'checksums': ['4e05d3d5307e6dde4c319187ae36e6cd3a0c035d4ed25e9e992449a304f47336'],
     }),
 ]
 
Diff against Optax-0.1.7-foss-2022a-CUDA-11.7.0.eb

easybuild/easyconfigs/o/Optax/Optax-0.1.7-foss-2022a-CUDA-11.7.0.eb

diff --git a/easybuild/easyconfigs/o/Optax/Optax-0.1.7-foss-2022a-CUDA-11.7.0.eb b/easybuild/easyconfigs/o/Optax/Optax-0.2.4-gfbf-2024a-CUDA-12.6.0.eb
index 51fa6c48d2..d8c53f9d2b 100644
--- a/easybuild/easyconfigs/o/Optax/Optax-0.1.7-foss-2022a-CUDA-11.7.0.eb
+++ b/easybuild/easyconfigs/o/Optax/Optax-0.2.4-gfbf-2024a-CUDA-12.6.0.eb
@@ -1,37 +1,30 @@
 easyblock = 'PythonBundle'
 
 name = 'Optax'
-version = '0.1.7'
+version = '0.2.4'
 versionsuffix = '-CUDA-%(cudaver)s'
 
 homepage = 'https://github.com/deepmind/optax'
 description = """Optax is a gradient processing and optimization library for JAX."""
 
-toolchain = {'name': 'foss', 'version': '2022a'}
+toolchain = {'name': 'gfbf', 'version': '2024a'}
 
 dependencies = [
-    ('CUDA', '11.7.0', '', SYSTEM),
-    ('Python', '3.10.4'),
-    ('dm-haiku', '0.0.9', versionsuffix),
-    ('dm-tree', '0.1.8'),
-    ('typing-extensions', '4.3.0'),
-]
-
-builddependencies = [
-    ('pytest-xdist', '2.5.0'),
+    ('CUDA', '12.6.0', '', SYSTEM),
+    ('Python', '3.12.3'),
+    ('SciPy-bundle', '2024.05'),
+    ('jax', '0.4.35', versionsuffix),
 ]
 
 exts_list = [
-    ('toolz', '0.12.0', {
-        'checksums': ['88c570861c440ee3f2f6037c4654613228ff40c93a6c25e0eba70d17282c6194'],
+    ('chex', '0.1.87', {
+        'checksums': ['0096d89cc8d898bb521ef4bfbf5c24549022b0e5b301f529ab57238896fe6c5d'],
     }),
-    ('chex', '0.1.6', {
-        'checksums': ['adb5d2352b5f0d248ccf594be1b1bf9ee7a2bee2a57f0eac78547538d479b0e7'],
+    ('toolz', '1.0.0', {
+        'checksums': ['2c86e3d9a04798ac556793bced838816296a2f085017664e4995cb40a1047a02'],
     }),
     ('optax', version, {
-        'checksums': ['6a5a848bc5e55e619b187c749fdddc4a5443ea14be85cc769f995779865c110d'],
-        # ignore equivalence_test.py which imports flax, which depends on optax
-        'runtest': 'python -m pytest -n %(parallel)s --pyargs --ignore-glob="**/equivalence_test.py" optax',
+        'checksums': ['4e05d3d5307e6dde4c319187ae36e6cd3a0c035d4ed25e9e992449a304f47336'],
     }),
 ]
 

@ThomasHoffmann77 ThomasHoffmann77 changed the title {lib}[foss/2024a] dm-haiku v0.0.13, tensorstore v0.1.69 w/ CUDA 12.6.0 {lib}[foss/2024a] dm-haiku v0.0.13 w/ CUDA 12.6.0 Mar 6, 2025
@Thyre Thyre added the 2024a issues & PRs related to 2024a common toolchains label Aug 18, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

2024a issues & PRs related to 2024a common toolchains update

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants