Skip to content

Add in_axes to vmap of transformation. #489

Add in_axes to vmap of transformation.

Add in_axes to vmap of transformation. #489