Skip to content

Commit

Permalink
adding easyconfigs: dm-haiku-0.0.13-foss-2024a-CUDA-12.6.0.eb, tensor…
Browse files Browse the repository at this point in the history
…store-0.1.69-foss-2024a.eb and patches: tensorstore-0.1.69_disable_third_party_downloads.patch
  • Loading branch information
ThomasHoffmann77 committed Dec 13, 2024
1 parent eefcf54 commit e280005
Show file tree
Hide file tree
Showing 3 changed files with 1,728 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,54 @@
# 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': '2024a'}

builddependencies = [
('poetry', '1.8.3')
]

dependencies = [
('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.69'),
('protobuf-python', '5.28.0'),
('Optax', '0.2.4', versionsuffix),
]

use_pip = True

exts_list = [
('jmp', '0.0.4', {
'checksums': ['5dfeb0fd7c7a9f72a70fff0aab9d0cbfae32a809c02f4037ff3485ceb33e1730'],
}),
('flax', '0.10.2', {
'checksums': ['6f831350026ad48182ba6588bb4dd72dc1084985d9aca923254cb3e4c78d75f3'],
}),
('nest_asyncio', '1.6.0', {
'checksums': ['6f172d5449aca15afd6c646851f4e31e02c598d553a667e38cafa997cfec55fe'],
}),
('humanize', '4.11.0'),
('orbax_checkpoint', '0.10.2', {
'modulename': 'orbax.checkpoint',
'checksums': ['e575ebe1f94e5cb6353ab8c9df81de0ca7cddc118645c3bfc17b8344f19d42f1'],
}),
('dm_haiku', version, {
'modulename': 'haiku',
'checksums': ['029bb91b5b1edb0d3fe23304d3bf12a545ea6e485041f7f5d8c8d85ebcf6e17d'],
}),
]

sanity_pip_check = True

moduleclass = 'lib'
Loading

0 comments on commit e280005

Please sign in to comment.