jax.experimental.global\_device\_array.GlobalDeviceArray ======================================================== .. currentmodule:: jax.experimental.global_device_array .. autoclass:: GlobalDeviceArray .. automethod:: __init__ .. rubric:: Methods .. autosummary:: ~GlobalDeviceArray.__init__ ~GlobalDeviceArray.from_batched_callback ~GlobalDeviceArray.from_batched_callback_with_devices ~GlobalDeviceArray.from_callback ~GlobalDeviceArray.local_data .. rubric:: Attributes .. autosummary:: ~GlobalDeviceArray.global_shards ~GlobalDeviceArray.is_fully_replicated ~GlobalDeviceArray.local_shards ~GlobalDeviceArray.ndim ~GlobalDeviceArray.shape ~GlobalDeviceArray.size