Skip to content

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