data-apis / array-api-compat

Compatibility layer for common array libraries to support the Array API
https://data-apis.org/array-api-compat/
MIT License
69 stars 22 forks source link
array-api cupy dask numpy pytorch

Array API compatibility library

This is a small wrapper around common array libraries that is compatible with the Array API standard. Currently, NumPy, CuPy, PyTorch, Dask, JAX, ndonnx 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/