Array API Standard Support: optimize.elementwise

Support on CPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	torch	jax	dask
find_root	✔️	✖	✖
find_minimum	✔️	✖	✖
bracket_root	✔️	✖	✖
bracket_minimum	✖	✖	✖

Support on GPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	cupy	torch	jax
find_root	✔️	✔️	✖
find_minimum	✔️	✔️	✖
bracket_root	✔️	✔️	✖
bracket_minimum	✔️	✖	✖

Support with JIT

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	jax
find_root	✖
find_minimum	✖
bracket_root	✖
bracket_minimum	✖