Any idea if the jax-metal package is under active development anymore?