Fix is_async_checkpointer now that AsyncManager is open-sourced from jax.experimental.
Fix is_async_checkpointer now that AsyncManager is open-sourced from jax.experimental.

PiperOrigin-RevId: 483507077

google/orbax