diff options
Diffstat (limited to 'share')
4 files changed, 12 insertions, 4 deletions
diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml index 0905305113..0526a4a12f 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-darwin-aarch64-mps/spack.yaml @@ -24,7 +24,9 @@ spack: # Keras # Bazel codesign issues - # - py-keras + # - py-keras backend=tensorflow + # - py-keras backend=jax + - py-keras backend=torch - py-keras-applications - py-keras-preprocessing - py-keras2onnx diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml index 71670d5a91..fdce7f9bb4 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cpu/spack.yaml @@ -19,7 +19,9 @@ spack: - py-jaxlib # Keras - - py-keras + # - py-keras backend=tensorflow + - py-keras backend=jax + - py-keras backend=torch - py-keras-applications - py-keras-preprocessing - py-keras2onnx diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml index 8829169038..1a64d83811 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml @@ -22,7 +22,9 @@ spack: - py-jaxlib # Keras - - py-keras + # - py-keras backend=tensorflow + - py-keras backend=jax + - py-keras backend=torch - py-keras-applications - py-keras-preprocessing - py-keras2onnx diff --git a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml index 620a95715b..5633d9846a 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-rocm/spack.yaml @@ -24,7 +24,9 @@ spack: - py-jaxlib # Keras - - py-keras + # - py-keras backend=tensorflow + - py-keras backend=jax + # - py-keras backend=torch - py-keras-applications - py-keras-preprocessing - py-keras2onnx |