jax.custom\_batching.custom\_vmap
=================================

.. currentmodule:: jax.custom_batching

.. autoclass:: custom_vmap

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~custom_vmap.__init__
      ~custom_vmap.def_vmap
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~custom_vmap.fun
      ~custom_vmap.vmap_rule