Multiple backends with array-api
With the work done in !2 (merged), it seems now relatively easy to support multiple backends for arrays, such as Numpy, Jax, CuPy.
What needs to be done:
- Implement a mixin, essentially in the same way as the Numpy mixin, which dispatches according to the backend
- Implement a hook that exposes the backend for each of the arithmetic classes