jax.experimental.global_device_array module
===========================================

.. automodule:: jax.experimental.global_device_array

API
---

.. autosummary::
   :toctree: _autosummary

   GlobalDeviceArray
   Shard