We should add the [Array API type hints](https://github.com/data-apis/array-api/tree/main/src/array_api_stubs) to the Cubed implementations. (Suggested by @TomNicholas.)