Skip to content

Fix filter_vmap with out_axes!=0,1 producing the wrong axis order. #367

Fix filter_vmap with out_axes!=0,1 producing the wrong axis order.

Fix filter_vmap with out_axes!=0,1 producing the wrong axis order. #367