Skip to content

{lib}[gfbf/2024a] dm-haiku v0.0.16#25472

Merged
branfosj merged 1 commit intoeasybuilders:developfrom
lexming:20260301114954_new_pr_dm-haiku0016
Mar 3, 2026
Merged

{lib}[gfbf/2024a] dm-haiku v0.0.16#25472
branfosj merged 1 commit intoeasybuilders:developfrom
lexming:20260301114954_new_pr_dm-haiku0016

Conversation

@lexming
Copy link
Contributor

@lexming lexming commented Mar 1, 2026

(created using eb --new-pr)

@lexming lexming added the update label Mar 1, 2026
@github-actions github-actions bot added the 2024a issues & PRs related to 2024a common toolchains label Mar 1, 2026
@github-actions
Copy link

github-actions bot commented Mar 1, 2026

Updated software dm-haiku-0.0.16-gfbf-2024a.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.16-gfbf-2024a.eb
index 9e73940419..b94fe20228 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.16-gfbf-2024a.eb
@@ -1,29 +1,27 @@
 easyblock = 'PythonBundle'
 
 name = 'dm-haiku'
-version = '0.0.13'
+version = '0.0.16'
 
 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': 'gfbf', 'version': '2024a'}
 
 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.6.2'),  # required by jmp, also provides absl-py
 ]
 
 exts_list = [
     ('jmp', '0.0.4', {
-        'source_tmpl': SOURCE_PY3_WHL,
-        'checksums': ['6aa7adbddf2bd574b28c7faf6e81a735eb11f53386447896909c6968dc36807d'],
+        'checksums': ['5dfeb0fd7c7a9f72a70fff0aab9d0cbfae32a809c02f4037ff3485ceb33e1730'],
     }),
-    ('dm_haiku', version, {
+    (name, version, {
         'modulename': 'haiku',
-        'source_tmpl': SOURCE_PY3_WHL,
-        'checksums': ['ee9562c68a059f146ad07f555ca591cb8c11ef751afecc38353863562bd23f43'],
+        'checksums': ['1830b0ce63c5cef2fb3a63a13033c9d8f612ee7f896f2b0b25a6ba484f5fad28'],
     }),
 ]
 
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.16-gfbf-2024a.eb
index 76672f6fb7..b94fe20228 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.16-gfbf-2024a.eb
@@ -1,45 +1,27 @@
-# update 0.0.12: Thomas Hoffmann (EMBL)
 easyblock = 'PythonBundle'
 
 name = 'dm-haiku'
-version = '0.0.12'
-versionsuffix = '-CUDA-%(cudaver)s'
+version = '0.0.16'
 
 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': 'gfbf', 'version': '2024a'}
 
 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.6.2'),  # required by jmp, also provides absl-py
 ]
 
 exts_list = [
     ('jmp', '0.0.4', {
         'checksums': ['5dfeb0fd7c7a9f72a70fff0aab9d0cbfae32a809c02f4037ff3485ceb33e1730'],
     }),
-    ('flax', '0.8.4', {
-        'checksums': ['968683f850198e1aa5eb2d9d1e20bead880ef7423c14f042db9d60848cb1c90b'],
-    }),
-    ('nest_asyncio', '1.6.0', {
-        'checksums': ['6f172d5449aca15afd6c646851f4e31e02c598d553a667e38cafa997cfec55fe'],
-    }),
-    ('orbax_checkpoint', '0.5.18', {
-        'modulename': 'orbax.checkpoint',
-        'preinstallopts': """sed -i 's/jax >= 0.4.25/&\\*/g' pyproject.toml &&""",
-        'checksums': ['29f5d311b412760bd6a2fecab3bdbf75407bc00dc6d0457d19478258ecc8fa6d'],
-    }),
     (name, version, {
         'modulename': 'haiku',
-        'checksums': ['ba0b3acf71433156737fe342c486da11727e5e6c9e054245f4f9b8f0b53eb608'],
+        'checksums': ['1830b0ce63c5cef2fb3a63a13033c9d8f612ee7f896f2b0b25a6ba484f5fad28'],
     }),
 ]
 
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.16-gfbf-2024a.eb
index eb44afa10f..b94fe20228 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.16-gfbf-2024a.eb
@@ -1,20 +1,18 @@
 easyblock = 'PythonBundle'
 
 name = 'dm-haiku'
-version = '0.0.9'
-versionsuffix = '-CUDA-%(cudaver)s'
+version = '0.0.16'
 
 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': 'gfbf', 'version': '2024a'}
 
 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.6.2'),  # required by jmp, also provides absl-py
 ]
 
 exts_list = [
@@ -23,7 +21,7 @@ exts_list = [
     }),
     (name, version, {
         'modulename': 'haiku',
-        'checksums': ['97752b32cdbe5a3e2d1c60ea884d33eb4b36e7d410000f0d61b571417c925435'],
+        'checksums': ['1830b0ce63c5cef2fb3a63a13033c9d8f612ee7f896f2b0b25a6ba484f5fad28'],
     }),
 ]
 

@lexming lexming mentioned this pull request Mar 1, 2026
1 task
@lexming
Copy link
Contributor Author

lexming commented Mar 2, 2026

@boegelbot please test @ jsc-zen3

@boegelbot
Copy link
Collaborator

@lexming: Request for testing this PR well received on jsczen3l1.int.jsc-zen3.fz-juelich.de

PR test command 'if [[ develop != 'develop' ]]; then EB_BRANCH=develop ./easybuild_develop.sh 2> /dev/null 1>&2; EB_PREFIX=/home/boegelbot/easybuild/develop source init_env_easybuild_develop.sh; fi; EB_PR=25472 EB_ARGS= EB_CONTAINER= EB_REPO=easybuild-easyconfigs EB_BRANCH=develop /opt/software/slurm/bin/sbatch --job-name test_PR_25472 --ntasks=8 ~/boegelbot/eb_from_pr_upload_jsc-zen3.sh' executed!

  • exit code: 0
  • output:
Submitted batch job 9919

Test results coming soon (I hope)...

Details

- notification for comment with ID 3983118880 processed

Message to humans: this is just bookkeeping information for me,
it is of no use to you (unless you think I have a bug, which I don't).

@boegelbot
Copy link
Collaborator

Test report by @boegelbot
SUCCESS
Build succeeded for 1 out of 1 (total: 1 min 13 secs) (1 easyconfigs in total)
jsczen3c1.int.jsc-zen3.fz-juelich.de - Linux Rocky Linux 9.7, x86_64, AMD EPYC-Milan Processor (zen3), Python 3.9.25
See https://gist.github.com/boegelbot/1d637268157496c1784c313cb749be52 for a full test report.

@branfosj branfosj added this to the next release (5.2.2?) milestone Mar 3, 2026
@branfosj
Copy link
Member

branfosj commented Mar 3, 2026

Test report by @branfosj
SUCCESS
Build succeeded for 1 out of 1 (total: 3 mins 36 secs) (1 easyconfigs in total)
bear-pg0210u03a - Linux RHEL 8.10, x86_64, Intel(R) Xeon(R) Platinum 8480CL (sapphirerapids), Python 3.6.8
See https://gist.github.com/branfosj/f23585b4cb5ab5ae32b5719ef51a122d for a full test report.

@branfosj
Copy link
Member

branfosj commented Mar 3, 2026

Going in, thanks @lexming!

@branfosj branfosj merged commit 42b3a17 into easybuilders:develop Mar 3, 2026
6 checks passed
@lexming lexming deleted the 20260301114954_new_pr_dm-haiku0016 branch March 3, 2026 10:51
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.

3 participants