mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 20:06:05 +00:00
13 lines
185 B
ReStructuredText
13 lines
185 B
ReStructuredText
``jax.experimental.shard_map`` module
|
|
=====================================
|
|
|
|
.. automodule:: jax.experimental.shard_map
|
|
|
|
API
|
|
---
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
shard_map
|