Array API Standard Support: cluster.vq

Support on CPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

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

Support on GPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

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

Support with JIT

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	jax
kmeans	✖
kmeans2	✖
vq	✖
whiten	✔️