Oven

array-api-compat1.9.1

Published 2 months ago

A wrapper around NumPy and other array libraries to make them compatible with the Array API standard

pip install array-api-compat

Package Downloads

Authors

Project URLs

Homepage

Requires Python

>=3.9

Dependencies

cupy; extra == "cupy"
dask; extra == "dask"
jax; extra == "jax"
numpy; extra == "numpy"
pytorch; extra == "pytorch"
sparse>=0.15.1; extra == "sparse"

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/