{ai}[foss/2024a] bitsandbytes v0.46.1, PyTorch-Lightning v2.5.2

easybuild/easyconfigs/b/bitsandbytes/bitsandbytes-0.46.1-foss-2024a.eb

@@ -0,0 +1,52 @@
+easyblock = 'CMakeMake'
+
+name = 'bitsandbytes'
+version = '0.46.1'
+
+homepage = 'https://huggingface.co/docs/bitsandbytes/main/en/index'
+description = "bitsandbytes enables accessible large language models via k-bit quantization for PyTorch."
+github_account = 'bitsandbytes-foundation'
+
+toolchain = {'name': 'foss', 'version': '2024a'}
+
+source_urls = [GITHUB_LOWER_SOURCE]
+sources = ['%(version)s.tar.gz']
+checksums = ['8326835082ad5590e4eab2cc51129bf55dd1c16e3d3038bc23431371c24b47da']
+
+builddependencies = [
+    ('CMake', '3.29.3'),
+    ('pkgconf', '2.2.0'),
+    ('scikit-build-core', '0.10.6'),
+]
+
+dependencies = [
+    ('Python', '3.12.3'),
+    ('PyTorch', '2.6.0'),
+    ('SciPy-bundle', '2024.05'),
+]
+
+configopts = "-DCOMPUTE_BACKEND=cpu"
+
+# skip install step in CMakeMake
+skipsteps = ['install']
+
+exts_defaultclass = 'PythonPackage'
+exts_list = [
+    (name, version, {
+        'source_urls': ['https://github.com/%(github_account)s/%(namelower)s/archive'],
+        'sources': ['%(version)s.tar.gz'],
+        'start_dir': '%(builddir)s/bitsandbytes-%(version)s/',
+        'checksums': ['8326835082ad5590e4eab2cc51129bf55dd1c16e3d3038bc23431371c24b47da'],
+    }),
+]
+
+sanity_check_paths = {
+    'files': [f'lib/python%(pyshortver)s/site-packages/bitsandbytes/libbitsandbytes_cpu.{SHLIB_EXT}'],
+    'dirs': ['lib/python%(pyshortver)s/site-packages'],
+}
+
+sanity_check_commands = [
+    "python -s -c 'import bitsandbytes'",
+]
+
+moduleclass = 'ai'

easybuild/easyconfigs/p/PyTorch-Lightning/PyTorch-Lightning-2.5.2-foss-2024a.eb

@@ -0,0 +1,40 @@
+easyblock = 'PythonBundle'
+
+name = 'PyTorch-Lightning'
+version = '2.5.2'
+
+homepage = 'https://pytorchlightning.ai'
+description = "PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers."
+
+toolchain = {'name': 'foss', 'version': '2024a'}
+
+dependencies = [
+    ('Python', '3.12.3'),
+    # dependencies from requirements/pytorch/base.txt
+    ('PyTorch', '2.6.0'),
+    ('tqdm', '4.66.5'),
+    ('aiohttp', '3.10.10'),
+    ('typing-extensions', '4.11.0'),
+    # optional dependencies from requirements/pytorch/extra.txt
+    ('bitsandbytes', '0.46.1'),
+    ('tensorboardX', '2.6.4'),
+    ('matplotlib', '3.9.2'),
+    ('Hydra', '1.3.2'),
+]
+
+# remaining dependencies from requirements/pytorch/base.txt
+exts_list = [
+    ('lightning-utilities', '0.14.3', {
+        'source_tmpl': 'lightning_utilities-%(version)s.tar.gz',
+        'checksums': ['37e2f83f273890052955a44054382c211a303012ee577619efbaa5df9e65e9f5'],
+    }),
+    ('torchmetrics', '1.7.3', {
+        'checksums': ['08450a19cdb67ba1608aac0b213e5dc73033e11b60ad4719696ebcede591621e'],
+    }),
+    ('pytorch-lightning', version, {
+        'source_tmpl': 'pytorch_lightning-%(version)s.tar.gz',
+        'checksums': ['f817087d611be8d43b777dd4e543d72703e235510936677a13e6c29f7fd790e3'],
+    }),
+]
+
+moduleclass = 'ai'