mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
35 lines
644 B
ReStructuredText
35 lines
644 B
ReStructuredText
``jax.sharding`` module
|
|
=======================
|
|
|
|
.. automodule:: jax.sharding
|
|
|
|
Classes
|
|
-------
|
|
|
|
.. currentmodule:: jax.sharding
|
|
|
|
.. autoclass:: Sharding
|
|
:members:
|
|
.. autoclass:: XLACompatibleSharding
|
|
:members:
|
|
:show-inheritance:
|
|
.. autoclass:: SingleDeviceSharding
|
|
:members:
|
|
:show-inheritance:
|
|
.. autoclass:: NamedSharding
|
|
:members:
|
|
:show-inheritance:
|
|
.. autoclass:: PositionalSharding
|
|
:members:
|
|
:show-inheritance:
|
|
.. autoclass:: PmapSharding
|
|
:members:
|
|
:show-inheritance:
|
|
.. autoclass:: GSPMDSharding
|
|
:members:
|
|
:show-inheritance:
|
|
.. autoclass:: PartitionSpec
|
|
:members:
|
|
.. autoclass:: Mesh
|
|
:members:
|