jax.experimental.custom\_dce.custom\_dce
========================================

.. currentmodule:: jax.experimental.custom_dce

.. autoclass:: custom_dce

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~custom_dce.__init__
      ~custom_dce.def_dce
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~custom_dce.fun
      ~custom_dce.static_argnums
      ~custom_dce.dce_rule