Skip to content

Latest commit

 

History

History
91 lines (87 loc) · 10.5 KB

api_status.md

File metadata and controls

91 lines (87 loc) · 10.5 KB

Array API Coverage Implementation Status

Cubed supports version 2022.12 of the Python array API standard, with a few exceptions noted below. The linear algebra extensions and Fourier transform functions¶ are not supported.

Support for version 2023.12 is tracked in Cubed issue #438.

This table shows which parts of the the Array API have been implemented in Cubed, and which ones are missing. The version column shows the version when the feature was added to the standard, for version 2022.12 or later.

Category Object/Function Implemented Version Notes
Array object Arithmetic Ops
Array Ops
Bitwise Ops
Comparison Ops
In-place Ops Arrays are immutable
Reflected Ops
Attributes
Methods Not device methods
Constants e, inf, ...
Creation Functions arange
asarray
empty
empty_like
eye
from_dlpack
full
full_like
linspace
meshgrid
ones
ones_like
tril
triu
zeros
zeros_like
Data Type Functions astype
can_cast
finfo
iinfo
result_type
Data Types bool, int8, ...
Elementwise Functions add Example of a binary function
negative Example of a unary function
others Except 2023.12 functions in #438
Indexing Single-axis
Multi-axis
Boolean array Shape is data dependent, #73
Indexing Functions take 2022.12
Inspection capabilities 2023.12
default_device 2023.12
default_dtypes 2023.12
devices 2023.12
dtypes 2023.12
Linear Algebra Functions matmul
matrix_transpose
tensordot
vecdot
Manipulation Functions broadcast_arrays
broadcast_to
concat
expand_dims
flip
permute_dims
repeat 2023.12
reshape Partial implementation
roll
squeeze
stack
tile 2023.12
unstack 2023.12
Searching Functions argmax
argmin
nonzero Shape is data dependent
searchsorted 2023.12
where
Set Functions unique_all Shape is data dependent
unique_counts Shape is data dependent
unique_inverse Shape is data dependent
unique_values Shape is data dependent
Sorting Functions argsort Not in Dask
sort Not in Dask
Statistical Functions cumulative_sum 2023.12
max
mean
min
prod
std Like mean, #29
sum
var Like mean, #29
Utility Functions all
any