brainstate.transform.vmap# brainstate.transform.vmap(fn=<brainstate.typing.Missing object>, *, in_axes=0, out_axes=0, axis_name=None, axis_size=None, spmd_axis_name=None, in_states=None, out_states=None)[source]# Return type: TypeVar(F, bound= Callable) | Callable[[TypeVar(F, bound= Callable)], TypeVar(F, bound= Callable)]
