diff options
Diffstat (limited to 'share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml')
-rw-r--r-- | share/spack/gitlab/cloud_pipelines/stacks/ml-linux-x86_64-cuda/spack.yaml | 4 |
1 files changed, 3 insertions, 1 deletions
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 |