4 ms·In addition, einsum in jax uses einops.by PartiallyTyped 5y agoIn addition, einsum in jax uses einops.