mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 12:56:07 +00:00
14 lines
167 B
ReStructuredText
14 lines
167 B
ReStructuredText
jax.experimental.maps module
|
|
============================
|
|
|
|
.. automodule:: jax.experimental.maps
|
|
|
|
API
|
|
---
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
Mesh
|
|
xmap
|