[jax_plugins]
metal = jax_plugins.metal
