Array API Standard Support: spatial

Support on CPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class	torch	jax	dask
ConvexHull	✖	✖	✖
Delaunay	✖	✖	✖
HalfspaceIntersection	✖	✖	✖
KDTree	✖	✖	✖
Rectangle	✖	✖	✖
SphericalVoronoi	✖	✖	✖
Voronoi	✖	✖	✖
cKDTree
convex_hull_plot_2d	N/A	N/A	N/A
delaunay_plot_2d	N/A	N/A	N/A
distance_matrix	N/A	N/A	N/A
geometric_slerp	✖	✖	✖
minkowski_distance	N/A	N/A	N/A
minkowski_distance_p	N/A	N/A	N/A
procrustes	✔️	✔️	✖
tsearch	N/A	N/A	N/A
voronoi_plot_2d	N/A	N/A	N/A

Support on GPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class	cupy	torch	jax
ConvexHull	✖	✖	✖
Delaunay	✖	✖	✖
HalfspaceIntersection	✖	✖	✖
KDTree	✖	✖	✖
Rectangle	✖	✖	✖
SphericalVoronoi	✖	✖	✖
Voronoi	✖	✖	✖
cKDTree
convex_hull_plot_2d	N/A	N/A	N/A
delaunay_plot_2d	N/A	N/A	N/A
distance_matrix	N/A	N/A	N/A
geometric_slerp	✖	✖	✖
minkowski_distance	N/A	N/A	N/A
minkowski_distance_p	N/A	N/A	N/A
procrustes	✔️	✔️	✔️
tsearch	N/A	N/A	N/A
voronoi_plot_2d	N/A	N/A	N/A

Support with JIT

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class	jax
ConvexHull	✖
Delaunay	✖
HalfspaceIntersection	✖
KDTree	✖
Rectangle	✖
SphericalVoronoi	✖
Voronoi	✖
cKDTree
convex_hull_plot_2d	N/A
delaunay_plot_2d	N/A
distance_matrix	N/A
geometric_slerp	✖
minkowski_distance	N/A
minkowski_distance_p	N/A
procrustes	✖
tsearch	N/A
voronoi_plot_2d	N/A