array-api-compat
A wrapper around NumPy and other array libraries to make them compatible with the Array API standard
Rank: #2622Downloads: 2,233,124 (30 days)
Description
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/