https://llego.dev/posts/numpy-vectorized-operations-broadcasting/