jax.vmap ======== .. currentmodule:: jax .. autofunction:: vmap