from einx._src.frontend.impl.jax import adapt_with_vmap
from einx._src.frontend.impl.jax import adapt_numpylike_reduce
from einx._src.frontend.impl.jax import adapt_numpylike_elementwise