array-api-compat
array-api-compat copied to clipboard
Compatibility layer for common array libraries to support the Array API
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, and JAX 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/