mirror of
https://github.com/ROCm/jax.git
synced 2025-04-25 02:16:05 +00:00
10 lines
198 B
ReStructuredText
10 lines
198 B
ReStructuredText
|
``jax.experimental.custom_partitioning`` module
|
||
|
|
===============================================
|
||
|
|
|
||
|
|
.. automodule:: jax.experimental.custom_partitioning
|
||
|
|
|
||
|
|
API
|
||
|
|
---
|
||
|
|
|
||
|
|
.. autofunction:: custom_partitioning
|