diff options
author | Adam J. Stewart <ajstewart426@gmail.com> | 2022-10-09 17:39:47 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-10-09 15:39:47 -0700 |
commit | 01ede3c595e8c02ecd791b97a0a1cae3ee30255f (patch) | |
tree | 9bbdd85946ee7f2bcb95c2781d8afd35db2cc161 /bin/sbang | |
parent | 4a6aff8bd1b3e1cacbdfcf7b1f4f95dbd6b8e3f9 (diff) | |
download | spack-01ede3c595e8c02ecd791b97a0a1cae3ee30255f.tar.gz spack-01ede3c595e8c02ecd791b97a0a1cae3ee30255f.tar.bz2 spack-01ede3c595e8c02ecd791b97a0a1cae3ee30255f.tar.xz spack-01ede3c595e8c02ecd791b97a0a1cae3ee30255f.zip |
Add CI stack for ML packages (#31592)
Basic stack of ML packages we would like to test and generate binaries for in CI.
Spack now has a large CI framework in GitLab for PR testing and public binary generation.
We should take advantage of this to test and distribute optimized binaries for popular ML
frameworks.
This is a pretty extensive initial set, including CPU, ROCm, and CUDA versions of a core
`x96_64_v4` stack.
### Core ML frameworks
These are all popular core ML frameworks already available in Spack.
- [x] PyTorch
- [x] TensorFlow
- [x] Scikit-learn
- [x] MXNet
- [x] CNTK
- [x] Caffe
- [x] Chainer
- [x] XGBoost
- [x] Theano
### ML extensions
These are domain libraries and wrappers that build on top of core ML libraries
- [x] Keras
- [x] TensorBoard
- [x] torchvision
- [x] torchtext
- [x] torchaudio
- [x] TorchGeo
- [x] PyTorch Lightning
- [x] torchmetrics
- [x] GPyTorch
- [x] Horovod
### ML-adjacent libraries
These are libraries that aren't specific to ML but are still core libraries used in ML pipelines
- [x] numpy
- [x] scipy
- [x] pandas
- [x] ONNX
- [x] bazel
Co-authored-by: Jonathon Anderson <17242663+blue42u@users.noreply.github.com>
Diffstat (limited to 'bin/sbang')
0 files changed, 0 insertions, 0 deletions