{ai,lib}[gfbf/2025b] Optax v0.2.5, Flax v0.10.4 by SimonPinches · Pull Request #25790 · easybuilders/easybuild-easyconfigs

SimonPinches · 2026-04-16T12:47:59Z

(created using eb --new-pr)

Depends upon:

jax/0.8.1: Fix jax version returned to not show .dev #25791

github-actions · 2026-04-16T12:48:29Z

Updated software `Flax-0.10.4-gfbf-2025b.eb`

Diff against Flax-0.8.4-gfbf-2023a-CUDA-12.1.1.eb

easybuild/easyconfigs/f/Flax/Flax-0.8.4-gfbf-2023a-CUDA-12.1.1.eb

diff --git a/easybuild/easyconfigs/f/Flax/Flax-0.8.4-gfbf-2023a-CUDA-12.1.1.eb b/easybuild/easyconfigs/f/Flax/Flax-0.10.4-gfbf-2025b.eb
index 802e02583f..7a6180b6b9 100644
--- a/easybuild/easyconfigs/f/Flax/Flax-0.8.4-gfbf-2023a-CUDA-12.1.1.eb
+++ b/easybuild/easyconfigs/f/Flax/Flax-0.10.4-gfbf-2025b.eb
@@ -1,39 +1,58 @@
 easyblock = 'PythonBundle'
 
 name = 'Flax'
-version = '0.8.4'
-versionsuffix = '-CUDA-%(cudaver)s'
+version = '0.10.4'
 
 homepage = 'https://flax.readthedocs.io'
 description = """Flax is a high-performance neural network library and ecosystem for JAX that is
 designed for flexibility: Try new forms of training by forking an example and
 by modifying the training loop, not by adding features to a framework."""
 
-toolchain = {'name': 'gfbf', 'version': '2023a'}
+toolchain = {'name': 'gfbf', 'version': '2025b'}
+
+builddependencies = [
+    ('hatchling', '1.27.0'),
+]
 
 dependencies = [
-    ('CUDA', '12.1.1', '', SYSTEM),
-    ('Python', '3.11.3'),
-    ('SciPy-bundle', '2023.07'),
-    ('jax', '0.4.25', versionsuffix),
-    ('Optax', '0.2.2', versionsuffix),
-    ('protobuf-python', '4.24.0'),
-    ('PyYAML', '6.0'),
+    ('Python', '3.13.5'),
+    ('SciPy-bundle', '2025.07'),
+    ('jax', '0.8.1'),
+    ('Optax', '0.2.5'),
+    ('PyYAML', '6.0.2'),
 ]
 
 exts_list = [
-    ('nest_asyncio', '1.6.0', {
-        'checksums': ['6f172d5449aca15afd6c646851f4e31e02c598d553a667e38cafa997cfec55fe'],
+    ('treescope', '0.1.10', {
+        'checksums': ['20f74656f34ab2d8716715013e8163a0da79bdc2554c16d5023172c50d27ea95'],
     }),
-    ('orbax_checkpoint', '0.5.15', {
-        'modulename': 'orbax.checkpoint',
-        'checksums': ['15195e8d1b381b56f23a62a25599a3644f5d08655fa64f60bb1b938b8ffe7ef3'],
+    ('etils', '1.12.2', {
+        'checksums': ['c6b9e1f0ce66d1bbf54f99201b08a60ba396d3446d9eb18d4bc39b26a2e1a5ee'],
+    }),
+    ('aiofiles', '24.1.0', {
+        'checksums': ['22a075c9e5a3810f0c2e48f3008c94d68c65d763b9b03857924c99e57355166c'],
+    }),
+    ('protobuf', '6.31.1', {
+        'modulename': 'google.protobuf',
+        'checksums': ['d8cac4c982f0b957a4dc73a80e2ea24fab08e679c0de9deb835f4a12d69aca9a'],
     }),
-    ('tensorstore', '0.1.60', {
-        'checksums': ['88da8f1978982101b8dbb144fd29ee362e4e8c97fc595c4992d555f80ce62a79'],
+    ('humanize', '4.12.3', {
+        'checksums': ['8430be3a615106fdfceb0b2c1b41c4c98c6b0fc5cc59663a5539b111dd325fb0'],
+    }),
+    ('uvloop', '0.21.0', {
+        'sources': ['uvloop-0.21.0-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl'],
+        'checksums': ['f3df876acd7ec037a3d005b3ab85a7e4110422e4d9c1571d4fc89b0fc41b6816'],
+    }),
+    ('tensorstore', '0.1.76', {
+        'sources': ['tensorstore-0.1.76-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl'],
+        'checksums': ['20782f833bfa3c59dd3787f657388054c54ee0ab48dad181b360e3e5e81e4c4b'],
+    }),
+    ('orbax-checkpoint', '0.11.36', {
+        'modulename': 'orbax.checkpoint',
+        'checksums': ['60ed7084a9b79385fb5b9e4b05d98c2db6f5892d05ee8d82df680cfac1622312'],
     }),
-    ('flax', '0.8.4', {
-        'checksums': ['968683f850198e1aa5eb2d9d1e20bead880ef7423c14f042db9d60848cb1c90b'],
+    ('flax', version, {
+        'checksums': ['57ae44d3f111fc85cff9049adb9684ce8ebd44e87bd8ca776ed52422c2d85021'],
     }),
 ]

Diff against Flax-0.8.4-gfbf-2023a.eb

easybuild/easyconfigs/f/Flax/Flax-0.8.4-gfbf-2023a.eb

diff --git a/easybuild/easyconfigs/f/Flax/Flax-0.8.4-gfbf-2023a.eb b/easybuild/easyconfigs/f/Flax/Flax-0.10.4-gfbf-2025b.eb
index fe69e5c053..7a6180b6b9 100644
--- a/easybuild/easyconfigs/f/Flax/Flax-0.8.4-gfbf-2023a.eb
+++ b/easybuild/easyconfigs/f/Flax/Flax-0.10.4-gfbf-2025b.eb
@@ -1,37 +1,58 @@
 easyblock = 'PythonBundle'
 
 name = 'Flax'
-version = '0.8.4'
+version = '0.10.4'
 
 homepage = 'https://flax.readthedocs.io'
 description = """Flax is a high-performance neural network library and ecosystem for JAX that is
 designed for flexibility: Try new forms of training by forking an example and
 by modifying the training loop, not by adding features to a framework."""
 
-toolchain = {'name': 'gfbf', 'version': '2023a'}
+toolchain = {'name': 'gfbf', 'version': '2025b'}
+
+builddependencies = [
+    ('hatchling', '1.27.0'),
+]
 
 dependencies = [
-    ('Python', '3.11.3'),
-    ('SciPy-bundle', '2023.07'),
-    ('jax', '0.4.25'),
-    ('Optax', '0.2.2'),
-    ('protobuf-python', '4.24.0'),
-    ('PyYAML', '6.0'),
+    ('Python', '3.13.5'),
+    ('SciPy-bundle', '2025.07'),
+    ('jax', '0.8.1'),
+    ('Optax', '0.2.5'),
+    ('PyYAML', '6.0.2'),
 ]
 
 exts_list = [
-    ('nest_asyncio', '1.6.0', {
-        'checksums': ['6f172d5449aca15afd6c646851f4e31e02c598d553a667e38cafa997cfec55fe'],
+    ('treescope', '0.1.10', {
+        'checksums': ['20f74656f34ab2d8716715013e8163a0da79bdc2554c16d5023172c50d27ea95'],
     }),
-    ('orbax_checkpoint', '0.5.15', {
-        'modulename': 'orbax.checkpoint',
-        'checksums': ['15195e8d1b381b56f23a62a25599a3644f5d08655fa64f60bb1b938b8ffe7ef3'],
+    ('etils', '1.12.2', {
+        'checksums': ['c6b9e1f0ce66d1bbf54f99201b08a60ba396d3446d9eb18d4bc39b26a2e1a5ee'],
+    }),
+    ('aiofiles', '24.1.0', {
+        'checksums': ['22a075c9e5a3810f0c2e48f3008c94d68c65d763b9b03857924c99e57355166c'],
+    }),
+    ('protobuf', '6.31.1', {
+        'modulename': 'google.protobuf',
+        'checksums': ['d8cac4c982f0b957a4dc73a80e2ea24fab08e679c0de9deb835f4a12d69aca9a'],
     }),
-    ('tensorstore', '0.1.60', {
-        'checksums': ['88da8f1978982101b8dbb144fd29ee362e4e8c97fc595c4992d555f80ce62a79'],
+    ('humanize', '4.12.3', {
+        'checksums': ['8430be3a615106fdfceb0b2c1b41c4c98c6b0fc5cc59663a5539b111dd325fb0'],
+    }),
+    ('uvloop', '0.21.0', {
+        'sources': ['uvloop-0.21.0-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl'],
+        'checksums': ['f3df876acd7ec037a3d005b3ab85a7e4110422e4d9c1571d4fc89b0fc41b6816'],
+    }),
+    ('tensorstore', '0.1.76', {
+        'sources': ['tensorstore-0.1.76-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl'],
+        'checksums': ['20782f833bfa3c59dd3787f657388054c54ee0ab48dad181b360e3e5e81e4c4b'],
+    }),
+    ('orbax-checkpoint', '0.11.36', {
+        'modulename': 'orbax.checkpoint',
+        'checksums': ['60ed7084a9b79385fb5b9e4b05d98c2db6f5892d05ee8d82df680cfac1622312'],
     }),
-    ('flax', '0.8.4', {
-        'checksums': ['968683f850198e1aa5eb2d9d1e20bead880ef7423c14f042db9d60848cb1c90b'],
+    ('flax', version, {
+        'checksums': ['57ae44d3f111fc85cff9049adb9684ce8ebd44e87bd8ca776ed52422c2d85021'],
     }),
 ]

Updated software `Optax-0.2.5-gfbf-2025b.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.5-gfbf-2025b.eb
index 45b587bc0d..f2ec54a387 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.5-gfbf-2025b.eb
@@ -1,30 +1,39 @@
 easyblock = 'PythonBundle'
 
 name = 'Optax'
-version = '0.2.2'
-versionsuffix = '-CUDA-%(cudaver)s'
+version = '0.2.5'
 
 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': '2025b'}
+
+builddependencies = [
+    ('hatchling', '1.27.0'),
+]
 
 dependencies = [
-    ('CUDA', '12.1.1', '', SYSTEM),
-    ('Python', '3.11.3'),
-    ('SciPy-bundle', '2023.07'),
-    ('jax', '0.4.25', versionsuffix),
+    ('Python', '3.13.5'),
+    ('SciPy-bundle', '2025.07'),
+    ('jax', '0.8.1'),
 ]
 
 exts_list = [
-    ('toolz', '0.12.1', {
-        'checksums': ['ecca342664893f177a13dac0e6b41cbd8ac25a358e5f215316d43e2100224f4d'],
+    ('toolz', '1.1.0', {
+        'sources': ['toolz-1.1.0-py3-none-any.whl'],
+        'checksums': ['15ccc861ac51c53696de0a5d6d4607f99c210739caf987b5d2054f3efed429d8'],
+    }),
+    ('wadler_lindig', '0.1.3', {
+        'checksums': ['476fb7015135f714cef8f8eac7c44b164c8b993345e651a9b6f25b7b112440c9'],
+    }),
+    ('jaxtyping', '0.3.2', {
+        'checksums': ['f30483fac4b42e915db8ad2414a85c3b63284aa7d3c100b96b59f755cf4a86ad'],
     }),
-    ('chex', '0.1.86', {
-        'checksums': ['e8b0f96330eba4144659e1617c0f7a57b161e8cbb021e55c6d5056c7378091d1'],
+    ('chex', '0.1.90', {
+        'checksums': ['d3c375aeb6154b08f1cccd2bee4ed83659ee2198a6acf1160d2fe2e4a6c87b5c'],
     }),
     ('optax', version, {
-        'checksums': ['f09bf790ef4b09fb9c35f79a07594c6196a719919985f542dc84b0bf97812e0e'],
+        'checksums': ['b2e38c7aea376186deae758ba7a258e6ef760c6f6131e9e11bc561c65386d594'],
     }),
 ]

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.5-gfbf-2025b.eb
index 007b56ab55..f2ec54a387 100644
--- a/easybuild/easyconfigs/o/Optax/Optax-0.2.2-gfbf-2023a.eb
+++ b/easybuild/easyconfigs/o/Optax/Optax-0.2.5-gfbf-2025b.eb
@@ -1,28 +1,39 @@
 easyblock = 'PythonBundle'
 
 name = 'Optax'
-version = '0.2.2'
+version = '0.2.5'
 
 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': '2025b'}
+
+builddependencies = [
+    ('hatchling', '1.27.0'),
+]
 
 dependencies = [
-    ('Python', '3.11.3'),
-    ('SciPy-bundle', '2023.07'),
-    ('jax', '0.4.25'),
+    ('Python', '3.13.5'),
+    ('SciPy-bundle', '2025.07'),
+    ('jax', '0.8.1'),
 ]
 
 exts_list = [
-    ('toolz', '0.12.1', {
-        'checksums': ['ecca342664893f177a13dac0e6b41cbd8ac25a358e5f215316d43e2100224f4d'],
+    ('toolz', '1.1.0', {
+        'sources': ['toolz-1.1.0-py3-none-any.whl'],
+        'checksums': ['15ccc861ac51c53696de0a5d6d4607f99c210739caf987b5d2054f3efed429d8'],
+    }),
+    ('wadler_lindig', '0.1.3', {
+        'checksums': ['476fb7015135f714cef8f8eac7c44b164c8b993345e651a9b6f25b7b112440c9'],
+    }),
+    ('jaxtyping', '0.3.2', {
+        'checksums': ['f30483fac4b42e915db8ad2414a85c3b63284aa7d3c100b96b59f755cf4a86ad'],
     }),
-    ('chex', '0.1.86', {
-        'checksums': ['e8b0f96330eba4144659e1617c0f7a57b161e8cbb021e55c6d5056c7378091d1'],
+    ('chex', '0.1.90', {
+        'checksums': ['d3c375aeb6154b08f1cccd2bee4ed83659ee2198a6acf1160d2fe2e4a6c87b5c'],
     }),
     ('optax', version, {
-        'checksums': ['f09bf790ef4b09fb9c35f79a07594c6196a719919985f542dc84b0bf97812e0e'],
+        'checksums': ['b2e38c7aea376186deae758ba7a258e6ef760c6f6131e9e11bc561c65386d594'],
     }),
 ]

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.5-gfbf-2025b.eb
index 970a86cb00..f2ec54a387 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.5-gfbf-2025b.eb
@@ -1,40 +1,42 @@
 easyblock = 'PythonBundle'
 
 name = 'Optax'
-version = '0.1.7'
-versionsuffix = '-CUDA-%(cudaver)s'
+version = '0.2.5'
 
 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': '2025b'}
 
-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 = [
+    ('hatchling', '1.27.0'),
 ]
 
-builddependencies = [
-    ('pytest-xdist', '2.5.0'),
+dependencies = [
+    ('Python', '3.13.5'),
+    ('SciPy-bundle', '2025.07'),
+    ('jax', '0.8.1'),
 ]
 
 exts_list = [
-    ('toolz', '0.12.0', {
-        'checksums': ['88c570861c440ee3f2f6037c4654613228ff40c93a6c25e0eba70d17282c6194'],
+    ('toolz', '1.1.0', {
+        'sources': ['toolz-1.1.0-py3-none-any.whl'],
+        'checksums': ['15ccc861ac51c53696de0a5d6d4607f99c210739caf987b5d2054f3efed429d8'],
+    }),
+    ('wadler_lindig', '0.1.3', {
+        'checksums': ['476fb7015135f714cef8f8eac7c44b164c8b993345e651a9b6f25b7b112440c9'],
+    }),
+    ('jaxtyping', '0.3.2', {
+        'checksums': ['f30483fac4b42e915db8ad2414a85c3b63284aa7d3c100b96b59f755cf4a86ad'],
     }),
-    ('chex', '0.1.6', {
-        'checksums': ['adb5d2352b5f0d248ccf594be1b1bf9ee7a2bee2a57f0eac78547538d479b0e7'],
+    ('chex', '0.1.90', {
+        'checksums': ['d3c375aeb6154b08f1cccd2bee4ed83659ee2198a6acf1160d2fe2e4a6c87b5c'],
     }),
     ('optax', version, {
-        'checksums': ['6a5a848bc5e55e619b187c749fdddc4a5443ea14be85cc769f995779865c110d'],
-        # ignore equivalence_test.py which imports flax, which depends on optax
-        'runtest': 'python -s -m pytest -n %(parallel)s --pyargs --ignore-glob="**/equivalence_test.py" optax',
+        'checksums': ['b2e38c7aea376186deae758ba7a258e6ef760c6f6131e9e11bc561c65386d594'],
     }),
 ]
 
-sanity_check_commands = ["python -s -c 'from optax import GradientTransformation'"]
+sanity_check_commands = ["python -c 'from optax import GradientTransformation'"]
 
 moduleclass = 'lib'

adding easyconfigs: Optax-0.2.5-gfbf-2025b.eb, Flax-0.10.4-gfbf-2025b.eb

3271477

github-actions Bot added 2025b issues & PRs related to 2025b common toolchains update labels Apr 16, 2026

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

{ai,lib}[gfbf/2025b] Optax v0.2.5, Flax v0.10.4#25790

{ai,lib}[gfbf/2025b] Optax v0.2.5, Flax v0.10.4#25790
SimonPinches wants to merge 1 commit intoeasybuilders:developfrom
SimonPinches:20260416144755_new_pr_Optax025

SimonPinches commented Apr 16, 2026 •

edited

Loading

Uh oh!

github-actions Bot commented Apr 16, 2026

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

Conversation

SimonPinches commented Apr 16, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

github-actions Bot commented Apr 16, 2026

Updated software Flax-0.10.4-gfbf-2025b.eb

Updated software Optax-0.2.5-gfbf-2025b.eb

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

1 participant

SimonPinches commented Apr 16, 2026 •

edited

Loading

Updated software `Flax-0.10.4-gfbf-2025b.eb`

Updated software `Optax-0.2.5-gfbf-2025b.eb`