# __array_api_version__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.__array_api_version__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
__array_api_version__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# abs

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.abs.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
abs

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# acos

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.acos.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
acos

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# acosh

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.acosh.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
acosh

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# add

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.add.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
add

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# all

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.all.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
all
axis
keepdims

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# any

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.any.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
any
axis
keepdims

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# arange

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.arange.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
arange
start
stop
step
dtype
device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# argmax

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.argmax.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
argmax
axis
keepdims

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# argmin

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.argmin.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
argmin
axis
keepdims

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# argsort

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.argsort.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
argsort
axis
descending
stable

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__abs__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__abs__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__abs__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__add__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__add__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__add__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__and__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__and__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__and__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__array_namespace__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__array__namespace__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__array_namespace__
api_version

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__bool__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__bool__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__bool__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__complex__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__complex__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__complex__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__eq__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__eq__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__float__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__float__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__float__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__floordiv__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__floordiv__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__floordiv__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__ge__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__ge__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__ge__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__getitem__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__getitem__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__getitem__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__gt__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__gt__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__gt__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__index__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__index__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__index__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__int__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__int__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__int__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__invert__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__invert__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__invert__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__le__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__le__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__le__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__lshift__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__lshift__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__lshift__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__lt__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__lt__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__lt__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__matmul__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__matmul__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__matmul__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__mod__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__mod__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__mod__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__mul__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__mul__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__mul__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__ne__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__ne__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__ne__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__neg__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__neg__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__neg__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__or__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__or__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__or__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__pos__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__pos__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__pos__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__pow__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__pow__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__pow__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__rshift__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__rshift__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__rshift__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__setitem__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__setitem__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__setitem__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__sub__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__sub__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__sub__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__truediv__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__truediv__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__truediv__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.__xor__

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.__xor__.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.__xor__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.device

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.device.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.dtype

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.dtype.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.dtype

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array

Spec: https://data-apis.org/array-api/latest/API_specification/array_object.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.mT

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.mT.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.mT

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.ndim

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.ndim.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.ndim

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.shape

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.shape.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.shape

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.size

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.size.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.size

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.T

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.T.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.T

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# array.to_device

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.to_device.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
array.to_device
stream

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# asarray

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.asarray.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
asarray
dtype
device
copy

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# asin

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.asin.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
asin

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# asinh

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.asinh.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
asinh

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# astype

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.astype.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
astype
copy

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# atan

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.atan.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
atan

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# atan2

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.atan2.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
atan2

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# atanh

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.atanh.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
atanh

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# bitwise_and

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.bitwise_and.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
bitwise_and

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# bitwise_invert

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.bitwise_invert.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
bitwise_invert

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# bitwise_left_shift

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.bitwise_left_shift.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
bitwise_left_shift

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# bitwise_or

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.bitwise_or.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
bitwise_or

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# bitwise_right_shift

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.bitwise_right_shift.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
bitwise_right_shift

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# bitwise_xor

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.bitwise_xor.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
bitwise_xor

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# bool

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
bool
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# broadcast_arrays

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.broadcast_arrays.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
broadcast_arrays

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# broadcast_to

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.broadcast_to.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
broadcast_to
shape

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# can_cast

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.can_cast.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
can_cast

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# ceil

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.ceil.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
ceil

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# complex128

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
complex128
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# complex64

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
complex64
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# concat

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.concat.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
concat
axis

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# conj

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.conj.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
conj

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# cos

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.cos.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
cos

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# cosh

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.cosh.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
cosh

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# divide

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.divide.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
divide

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# e

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.e.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
e

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# empty

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.empty.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
empty
dtype
device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# empty_like

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.empty_like.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
empty_like
dtype
device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# equal

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.equal.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
equal

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# exp

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.exp.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
exp

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# expand_dims

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.expand_dims.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
expand_dims
axis

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# expm1

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.expm1.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
expm1

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# eye

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.eye.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
eye
k
dtype
device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.fft

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.fft.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.fft
n
axis
norm

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.fftfreq

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.fftfreq.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.fftfreq
d
device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.fftn

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.fftn.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.fftn
s
axes
norm

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.fftshift

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.fftshift.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.fftshift
axes

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.hfft

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.hfft.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.hfft
n
axis
norm

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.ifft

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.ifft.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.ifft
n
axis
norm

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.ifftn

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.ifftn.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.ifftn
s
axes
norm

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.ifftshift

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.ifftshift.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.ifftshift
axes

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.ihfft

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.ihfft.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.ihfft
n
axis
norm

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.irfft

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.irfft.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.irfft
n
axis
norm

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.irfftn

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.irfftn.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.irfftn
s
axes
norm

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.rfft

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.rfft.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.rfft
n
axis
norm

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.rfftfreq

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.rfftfreq.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.rfftfreq
d
device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# fft.rfftn

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.fft.rfftn.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
fft.rfftn
s
axes
norm

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# finfo

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.finfo.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
finfo
bits
eps
max
min
smallest_normal
dtype

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# flip

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.flip.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
flip
axis

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# float32

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
float32
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# float64

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
float64
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# floor

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.floor.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
floor

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# floor_divide

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.floor_divide.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
floor_divide

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# full

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.full.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
full
dtype
device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# full_like

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.full_like.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
full_like
dtype
device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# greater

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.greater.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
greater

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# greater_equal

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.greater_equal.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
greater_equal

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# iinfo

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.iinfo.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
iinfo
bits
max
min
dtype

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# imag

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.imag.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
imag

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# Array Indexing

Spec: https://data-apis.org/array-api/latest/API_specification/generated/indexing.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
Array Indexing
Single axis
Multiple axis
Boolean

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# inf

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.inf.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
inf

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# int16

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
int16
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# int32

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
int32
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# int64

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
int64
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# int8

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
int8
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# isdtype

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.isdtype.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
isdtype

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# isfinite

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.isfinite.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
isfinite

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# isinf

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.isinf.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
isinf

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# isnan

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.isnan.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
isnan

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# less

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.less.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
less

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# less_equal

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.less_equal.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
less_equal

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.cholesky

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.cholesky.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.cholesky
upper

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.cross

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.cross.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.cross
axis

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.det

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.det.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.det

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.diagonal

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.diagonal.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.diagonal
offset

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.eigh

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.eigh.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.eigh

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.eigvalsh

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.eigvalsh.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.eigvalsh

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.inv

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.inv.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.inv

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.matmul

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.matmul.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.matmul

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.matrix_norm

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.matrix_norm.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.matrix_norm
keepdims
ord

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.matrix_power

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.matrix_power.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.matrix_power

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.matrix_rank

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.matrix_rank.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.matrix_rank
rtol

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.matrix_transpose

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.matrix_transpose.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.matrix_transpose

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.outer

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.outer.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.outer

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.pinv

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.pinv.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.pinv
rtol

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.qr

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.qr.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.qr
mode

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.slogdet

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.slogdet.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.slogdet

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.solve

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.solve.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.solve

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.svd

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.svd.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.svd
full_matrices

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.svdvals

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.svdvals.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.svdvals

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.trace

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.trace.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.trace
offset
dtype

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.vecdot

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.vecdot.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.vecdot
axis

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linalg.vector_norm

Spec: https://data-apis.org/array-api/latest/extensions/generated/array_api.linalg.vector_norm.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linalg.vector_norm
axis
keepdims
ord

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# linspace

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.linspace.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
linspace
num
dtype
device
endpoint

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# log

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.log.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
log

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# log10

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.log10.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
log10

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# log1p

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.log1p.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
log1p

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# log2

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.log2.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
log2

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# logaddexp

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.logaddexp.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
logaddexp

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# logical_and

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.logical_and.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
logical_and

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# logical_not

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.logical_not.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
logical_not

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# logical_or

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.logical_or.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
logical_or

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# logical_xor

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.logical_xor.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
logical_xor

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# matmul

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.matmul.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
matmul

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# matrix_transpose

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.matrix_transpose.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
matrix_transpose

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# max

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.max.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
max
axis
keepdims

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# mean

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.mean.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
mean
axis
keepdims

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# meshgrid

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.meshgrid.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
meshgrid
indexing

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# min

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.min.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
min
axis
keepdims

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# multiply

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.multiply.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
multiply

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# nan

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.nan.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
nan

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# negative

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.negative.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
negative

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# newaxis

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.newaxis.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
newaxis

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# nonzero

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.nonzero.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
nonzero

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# not_equal

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.not_equal.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
not_equal

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# ones

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.ones.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
ones
dtype
device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# ones_like

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.ones_like.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
ones_like
dtype
device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# permute_dims

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.permute_dims.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
permute_dims
axes

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# pi

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.pi.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
pi

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# positive

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.positive.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
positive

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# pow

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.pow.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
pow

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# prod

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.prod.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
prod
axis
dtype
keepdims

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# real

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.real.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
real

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# remainder

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.remainder.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
remainder

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# reshape

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.reshape.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
reshape
shape
copy

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# result_type

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.result_type.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
result_type

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# roll

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.roll.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
roll
shift
axis

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# round

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.round.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
round

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# sign

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.sign.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
sign

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# sin

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.sin.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
sin

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# sinh

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.sinh.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
sinh

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# sort

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.sort.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
sort
axis
descending
stable

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# sqrt

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.sqrt.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
sqrt

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# square

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.square.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
square

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# squeeze

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.squeeze.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
squeeze
axis

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# stack

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.stack.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
stack
axis

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# std

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.std.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
std
axis
correction
keepdims

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# subtract

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.subtract.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
subtract

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# sum

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.sum.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
sum
axis
dtype
keepdims

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# take

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.take.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
take
axis

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# tan

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.tan.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
tan

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# tanh

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.tanh.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
tanh

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# tensordot

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.tensordot.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
tensordot
axes

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# tril

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.tril.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
tril
k

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# triu

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.triu.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
triu
k

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# trunc

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.trunc.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
trunc

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# uint16

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
uint16
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# uint32

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
uint32
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# uint64

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
uint64
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# uint8

Spec: https://data-apis.org/array-api/latest/API_specification/generated/data_types.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
uint8
__eq__

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# var

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.var.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
var
axis
correction
keepdims

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# vecdot

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.vecdot.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
vecdot
axis

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# where

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.where.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
where

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# zeros

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.zeros.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
zeros
dtype
device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support

# zeros_like

Spec: https://data-apis.org/array-api/latest/API_specification/generated/array_api.zeros_like.html

CuPy
Dask
JAX
MXNet
NumPy
PyTorch
TensorFlow
zeros_like
dtype
device

Legend

Tip: you can click/tap on a cell for more information.

Full support
Full support
Partial support
Partial support
Preview support
Experimental
No support
No support