Also needs that in some of our internal dependencies (see sub-issues). IIRC, there are a few cases like `np.trapz` in the codebase that need to be changed, shouldn't take long.