Add jax.make_array_from_process_local_data to the docs. Preview: https://jax--21444.org.readthedocs.build/en/21444/_autosummary/jax.make_array_from_process_local_data.html#jax.make_array_from_process_local_data

PiperOrigin-RevId: 637660828

Author: yashk2810

docs/jax.rst

@@ -113,6 +113,7 @@ jax.Array (:code:`jax.Array`)
     Array
     make_array_from_callback
     make_array_from_single_device_arrays
+    make_array_from_process_local_data
 
 Vectorization (:code:`vmap`)
 ----------------------------