einx.jax.adapt\_with\_vmap
==========================

.. currentmodule:: einx.jax

.. autofunction:: adapt_with_vmap