brainstate.transform.vmap2
==========================

.. currentmodule:: brainstate.transform

.. autofunction:: vmap2