|
2 | 2 |
|
3 | 3 | This is a small wrapper around common array libraries that is compatible with
|
4 | 4 | the [Array API standard](https://data-apis.org/array-api/latest/). Currently,
|
5 |
| -NumPy, CuPy, PyTorch, Dask, JAX, and Sparse are supported. If you want support |
6 |
| -for other array libraries, or if you encounter any issues, please [open an |
7 |
| -issue](https://github.com/data-apis/array-api-compat/issues). |
| 5 | +NumPy, CuPy, PyTorch, Dask, JAX, ndonnx, and Sparse are supported. If you want |
| 6 | +support for other array libraries, or if you encounter any issues, please |
| 7 | +[open an issue](https://github.com/data-apis/array-api-compat/issues). |
8 | 8 |
|
9 | 9 | Note that some of the functionality in this library is backwards incompatible
|
10 | 10 | with the corresponding wrapped libraries. The end-goal is to eventually make
|
@@ -61,10 +61,11 @@ import array_api_compat.dask as da
|
61 | 61 | ```
|
62 | 62 |
|
63 | 63 | ```{note}
|
64 |
| -There is no `array_api_compat.jax` submodule. JAX support is contained in JAX |
65 |
| -itself in the `jax.experimental.array_api` module. array-api-compat simply |
66 |
| -wraps that submodule. The main JAX support in this module consists of |
67 |
| -supporting it in the [helper functions](helper-functions). |
| 64 | +There are no `array_api_compat` submodules for JAX, sparse, or ndonnx. These |
| 65 | +support for these libraries is contained in the libraries themselves (JAX |
| 66 | +support is in the `jax.experimental.array_api` module). The |
| 67 | +array-api-compat support for these libraries consists of supporting them in |
| 68 | +the [helper functions](helper-functions). |
68 | 69 | ```
|
69 | 70 |
|
70 | 71 | Each will include all the functions from the normal NumPy/CuPy/PyTorch/dask.array
|
|
0 commit comments