Skip to content

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

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. #1017