[jax_plugins]
xla_cuda12 = jax_plugins.xla_cuda12