[jax_plugins]
xla_rocm1 = jax_plugins.xla_rocm1
