(And their implementation of `axis` involves Python function calls, so it's not vectorized)