Support jax.debug.callback() in MPMD.

All we need to do is to properly pass down the host_callbacks to IFRT compiler.

PiperOrigin-RevId: 965225367