[jax_plugins]
mtl = metal_pjrt_plugin
