diff --git a/easybuild/easyconfigs/a/Ax/Ax-0.3.3-foss-2022a.eb b/easybuild/easyconfigs/a/Ax/Ax-0.3.3-foss-2022a.eb new file mode 100644 index 00000000000..e9e66b25092 --- /dev/null +++ b/easybuild/easyconfigs/a/Ax/Ax-0.3.3-foss-2022a.eb @@ -0,0 +1,53 @@ +easyblock = 'PythonBundle' + +name = 'Ax' +version = '0.3.3' + +homepage = 'https://ax.dev/' +description = """ +Ax is an accessible, general-purpose platform for understanding, managing, deploying, and +automating adaptive experiments. +Adaptive experimentation is the machine-learning guided process of iteratively exploring +a (possibly infinite) parameter space in order to identify optimal configurations in a +resource-efficient manner. Ax currently supports Bayesian optimization and bandit +optimization as exploration strategies. Bayesian optimization in Ax is powered by +BoTorch, a modern library for Bayesian optimization research built on PyTorch. +""" + +toolchain = {'name': 'foss', 'version': '2022a'} + +dependencies = [ + ('Python', '3.10.4'), + ('SciPy-bundle', '2022.05'), + ('PyTorch', '1.12.0'), + ('scikit-learn', '1.1.2'), + ('IPython', '8.5.0'), + ('plotly.py', '5.12.0'), + ('Parsl', '2023.7.17'), + ('pyro-ppl', '1.8.4'), +] + +use_pip = True + +exts_list = [ + ('multipledispatch', '1.0.0', { + 'checksums': ['5c839915465c68206c3e9c473357908216c28383b425361e5d144594bf85a7e0'], + }), + ('linear_operator', '0.4.0', { + 'checksums': ['7c57c9f8f258c9785c0db4dd7625f4dd03a340313d7314cba0b633644909f5c6'], + }), + ('gpytorch', '1.10', { + 'checksums': ['6dc978ab9fbf220a845a4f1ea13104180fc50e6934081f421b37f6120afb7f18'], + }), + ('botorch', '0.8.5', { + 'checksums': ['8598e3b8a0da26ed7e21a364486c0ae1188de63183d36874652fad39084c5848'], + }), + ('ax-platform', version, { + 'modulename': 'ax', + 'checksums': ['4f067d2b29e8352e826074b9360b5b1ed9f9e17bd49ac01840cf25111c621d1f'], + }), +] + +sanity_pip_check = True + +moduleclass = 'ai' diff --git a/easybuild/easyconfigs/p/pyro-ppl/pyro-ppl-1.8.4-foss-2022a.eb b/easybuild/easyconfigs/p/pyro-ppl/pyro-ppl-1.8.4-foss-2022a.eb new file mode 100644 index 00000000000..05afed8446b --- /dev/null +++ b/easybuild/easyconfigs/p/pyro-ppl/pyro-ppl-1.8.4-foss-2022a.eb @@ -0,0 +1,44 @@ +# Author: Denis Krišťák (INUITS) + +easyblock = 'PythonBundle' + +name = 'pyro-ppl' +version = '1.8.4' + +homepage = 'https://github.com/pyro-ppl/pyro' +description = "Pyro is a flexible, scalable deep probabilistic programming library built on PyTorch." + +toolchain = {'name': 'foss', 'version': '2022a'} + +dependencies = [ + ('Python', '3.10.4'), + ('SciPy-bundle', '2022.05'), + ('PyTorch', '1.12.0'), + ('tqdm', '4.64.0'), +] + +use_pip = True + +exts_list = [ + ('opt_einsum', '3.3.0', { + 'checksums': ['59f6475f77bbc37dcf7cd748519c0ec60722e91e63ca114e68821c0c54a46549'], + }), + ('pyro-api', '0.1.2', { + 'modulename': 'pyroapi', + 'checksums': ['a1b900d9580aa1c2fab3b123ab7ff33413744da7c5f440bd4aadc4d40d14d920'], + }), + (name, version, { + 'modulename': 'pyro', + 'checksums': ['766fad61e52df48885de96d41213da1f8e8c1b79ecf308ad53189fcd15c1cb41'], + }), +] + +sanity_pip_check = True + +sanity_check_commands = [ + "python -c 'from pyroapi import distributions as dist'", + "python -c 'from pyroapi import infer, ops, optim, pyro, pyro_backend'", + "python -c 'from pyro import infer, nn, distributions'", +] + +moduleclass = 'tools'