3 ms·But also jax.numpy can do this quite a bit too.by N1H1L 5y agoBut also jax.numpy can do this quite a bit too.