https://blog.ando.ai/posts/matmul-jacobian/