summaryrefslogtreecommitdiff
path: root/README.md
diff options
context:
space:
mode:
authorAdam J. Stewart <ajstewart426@gmail.com>2022-10-09 17:39:47 -0500
committerGitHub <noreply@github.com>2022-10-09 15:39:47 -0700
commit01ede3c595e8c02ecd791b97a0a1cae3ee30255f (patch)
tree9bbdd85946ee7f2bcb95c2781d8afd35db2cc161 /README.md
parent4a6aff8bd1b3e1cacbdfcf7b1f4f95dbd6b8e3f9 (diff)
downloadspack-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 'README.md')
0 files changed, 0 insertions, 0 deletions