diff options
Diffstat (limited to 'share')
3 files changed, 6 insertions, 9 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 d4e86fd501..b5d989c904 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-cpu/spack.yaml @@ -28,9 +28,8 @@ spack: - py-transformers # JAX - # https://github.com/google/jax/issues/12614 - # - py-jax - # - py-jaxlib + - py-jax + - py-jaxlib # Keras - py-keras 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 f04b96ce93..1ea78372cb 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-cuda/spack.yaml @@ -31,9 +31,8 @@ spack: - py-transformers # JAX - # https://github.com/google/jax/issues/12614 - # - py-jax - # - py-jaxlib + - py-jax + - py-jaxlib # Keras - py-keras 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 4c44e881d7..2d728b501e 100644 --- a/share/spack/gitlab/cloud_pipelines/stacks/ml-rocm/spack.yaml +++ b/share/spack/gitlab/cloud_pipelines/stacks/ml-rocm/spack.yaml @@ -33,9 +33,8 @@ spack: - py-transformers # JAX - # https://github.com/google/jax/issues/12614 - # - py-jax - # - py-jaxlib + - py-jax + - py-jaxlib # Keras - py-keras |