This is a small wrapper around common array libraries that is compatible with
the Array API standard. Currently,
NumPy, CuPy, PyTorch, Dask, JAX and sparse
are supported. If you want support
for other array libraries, or if you encounter any issues, please open an
issue.
See the documentation for more details https://data-apis.org/array-api-compat/