mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 21:06:06 +00:00
e6c4d4a30e
Also moving jax_array_migration guide to reference documentation. PiperOrigin-RevId: 488489503
22 lines
367 B
ReStructuredText
22 lines
367 B
ReStructuredText
jax.sharding package
|
|
====================
|
|
|
|
.. automodule:: jax.sharding
|
|
|
|
Classes
|
|
-------
|
|
|
|
.. currentmodule:: jax.sharding
|
|
|
|
.. autoclass:: Sharding
|
|
:members:
|
|
.. autoclass:: XLACompatibleSharding
|
|
:members:
|
|
:show-inheritance:
|
|
.. autoclass:: NamedSharding
|
|
:members:
|
|
:show-inheritance:
|
|
.. autoclass:: SingleDeviceSharding
|
|
:members:
|
|
:show-inheritance:
|