ML CI: reenable JAX builds on Darwin (spack#38)

Author: adamjstewart

stacks/ml-darwin-aarch64-mps/spack.yaml

@@ -26,14 +26,12 @@ spack:
   - py-transformers
 
   # JAX
-  # Bazel codesign issues
-  # - py-jax
-  # - py-jaxlib
+  - py-jax
+  - py-jaxlib
 
   # Keras
-  # Bazel codesign issues
   # - py-keras backend=tensorflow
-  # - py-keras backend=jax
+  - py-keras backend=jax
   - py-keras backend=torch
 
   # PyTorch
@@ -62,7 +60,6 @@ spack:
   - py-tensorboardx
 
   # TensorFlow
-  # Bazel codesign issues
   # - py-tensorflow
   # - py-tensorflow-datasets
   # - py-tensorflow-metadata