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

.. currentmodule:: jax.custom_batching

.. automethod:: custom_vmap.def_vmap