3 ms·Ah right, yes I remember. Perhaps .vmap from pytorch?by PartiallyTyped 6y agoAh right, yes I remember. Perhaps .vmap from pytorch?