mirror of
https://github.com/ROCm/jax.git
synced 2025-04-17 12:26:07 +00:00
12 lines
220 B
ReStructuredText
12 lines
220 B
ReStructuredText
jax.experimental.global_device_array module
|
|
===========================================
|
|
|
|
.. automodule:: jax.experimental.global_device_array
|
|
|
|
API
|
|
---
|
|
|
|
.. autoclass:: GlobalDeviceArray
|
|
:members:
|
|
.. autoclass:: Shard
|