Jax
---

..  autosummary::
    :toctree: jax
    :nosignatures:

    einx.jax.adapt_with_vmap
    einx.jax.adapt_numpylike_reduce
    einx.jax.adapt_numpylike_elementwise