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

.. automodule:: jax.experimental.global_device_array

API
---

.. autosummary::
   :toctree: _autosummary

   GlobalDeviceArray
   Shard
jax.experimental.global_device_array module
===========================================

.. automodule:: jax.experimental.global_device_array

API
---

.. autoclass:: GlobalDeviceArray
    :members:
.. autoclass:: Shard

Fix the display of GDA docs
Fix the display of GDA docs
