jax.experimental.custom_dce module

.. automodule:: jax.experimental.custom_dce

API

.. autosummary:: :toctree: _autosummary

custom_dce custom_dce.def_dce