Array API Standard Support: differentiate

Support on CPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	torch	jax	dask
derivative	✔️	✔️	✖
jacobian	✔️	✔️	✖
hessian	✔️	✔️	✖

Support on GPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	cupy	torch	jax
derivative	✔️	✔️	✔️
jacobian	✔️	✔️	✔️
hessian	✔️	✔️	✔️

Support with JIT

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	jax
derivative	✖
jacobian	✖
hessian	✖