I'm also interested in this. Apple Silicon is an exciting hardware platform for ML and we need guidance from Apple how it envisions the future ecosystem. Is there going to be more openness and support of the major ML frameworks (incl. JAX), or is jax-metal an abandoned tech demo and we're all meant to use MLX more or less exclusively? The latter scenario would be incredibly unfortunate.