Hi is Jax for Metal or apple silicon is supported any more it seems the backend metal is still not officially supported by JAX as per there repo.