> features that numpy can't really do like native GPU execution

we use numpy + jax for that; works well