mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +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
|