diff options
Diffstat (limited to 'share')
3 files changed, 9 insertions, 0 deletions
diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml index a03ecbae69..6b6280d66c 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml @@ -23,6 +23,9 @@ spack: # Horovod - py-horovod + # Hugging Face + - py-transformers + # JAX # https://github.com/google/jax/issues/12614 # - py-jax diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml index ada465fd1a..9d050b5b01 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml @@ -26,6 +26,9 @@ spack: # Horovod - py-horovod + # Hugging Face + - py-transformers + # JAX # https://github.com/google/jax/issues/12614 # - py-jax diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-rocm/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-rocm/spack.yaml index bbc6159f59..f1c5fa3956 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-rocm/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-rocm/spack.yaml @@ -28,6 +28,9 @@ spack: # Horovod - py-horovod + # Hugging Face + - py-transformers + # JAX # https://github.com/google/jax/issues/12614 # - py-jax |