File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change 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, and PyTorch are supported. If you want support for other array
5
+ NumPy, CuPy, PyTorch, and Dask are supported. If you want support for other array
6
6
libraries, or if you encounter any issues, please [ open an
7
7
issue] ( https://github.com/data-apis/array-api-compat/issues ) .
8
8
@@ -56,7 +56,11 @@ import array_api_compat.cupy as cp
56
56
import array_api_compat.torch as torch
57
57
```
58
58
59
- Each will include all the functions from the normal NumPy/CuPy/PyTorch
59
+ ``` py
60
+ import array_api_compat.dask as da
61
+ ```
62
+
63
+ Each will include all the functions from the normal NumPy/CuPy/PyTorch/dask.array
60
64
namespace, except that functions that are part of the array API are wrapped so
61
65
that they have the correct array API behavior. In each case, the array object
62
66
used will be the same array object from the wrapped library.
You can’t perform that action at this time.
0 commit comments