Array API Standard Support: cluster.vq

Support on CPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	torch	jax	dask
whiten	✔️	✔️	✔️
vq	✔️	✔️	✔️
kmeans	✔️	✔️	✔️
kmeans2	✔️	✔️	✔️

Support on GPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	cupy	torch	jax
whiten	✔️	✔️	✔️
vq	✖	✖	✖
kmeans	✖	✖	✖
kmeans2	✖	✖	✖

Support with JIT

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	jax
whiten	✔️
vq	✖
kmeans	✖
kmeans2	✖