mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 03:46:06 +00:00
14 lines
210 B
ReStructuredText
14 lines
210 B
ReStructuredText
|
``jax.experimental.custom_dce`` module
|
||
|
|
======================================
|
||
|
|
|
||
|
|
.. automodule:: jax.experimental.custom_dce
|
||
|
|
|
||
|
|
API
|
||
|
|
---
|
||
|
|
|
||
|
|
.. autosummary::
|
||
|
|
:toctree: _autosummary
|
||
|
|
|
||
|
|
custom_dce
|
||
|
|
custom_dce.def_dce
|