Add `host_local_array_to_global_array` and `global_array_to_host_local_array` for enabling transition to jax.Array.
Add `host_local_array_to_global_array` and `global_array_to_host_local_array` for enabling transition to jax.Array.

Also support `FROM_GDA` for `jax.Array` as a backwards compatible change so that users can continue to use that until they transition to jax.Array. Its currently required because of usage like `in_axis_resources = (FROM_GDA, FROM_GDA, P('data'), None)` and changing this on users side will require input from users so we as JAX can just support it as a temporary thing since GDA and Array act similarly in pjit.

PiperOrigin-RevId: 479035326

google/jax,google/jax,google/jax,google/jax