Array API Standard Support: integrate

Support on CPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	torch	jax	dask
cubature	✔️	✔️	✔️
cumulative_simpson	✔️	✖	✔️
cumulative_trapezoid	✔️	✔️	✔️
dblquad	✖	✖	✖
fixed_quad	✖	✖	✖
lebedev_rule	✖	✖	✖
newton_cotes	✖	✖	✖
nquad	✖	✖	✖
nsum	✖	✖	✖
odeint	N/A	N/A	N/A
qmc_quad	✔️	✔️	✖
quad	✖	✖	✖
quad_vec	✖	✖	✖
romb	✔️	✔️	✔️
simpson	✖	✖	✖
solve_bvp	✖	✖	✖
solve_ivp	✖	✖	✖
tanhsinh	✔️	✖	✖
tplquad	✖	✖	✖
trapezoid	✔️	✖	✔️

Support on GPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	cupy	torch	jax
cubature	✔️	✔️	✔️
cumulative_simpson	✔️	✔️	✖
cumulative_trapezoid	✔️	✔️	✔️
dblquad	✖	✖	✖
fixed_quad	✖	✖	✖
lebedev_rule	✖	✖	✖
newton_cotes	✖	✖	✖
nquad	✖	✖	✖
nsum	✔️	✖	✖
odeint	N/A	N/A	N/A
qmc_quad	✔️	✔️	✔️
quad	✖	✖	✖
quad_vec	✖	✖	✖
romb	✔️	✔️	✔️
simpson	✖	✖	✖
solve_bvp	✖	✖	✖
solve_ivp	✖	✖	✖
tanhsinh	✔️	✔️	✖
tplquad	✖	✖	✖
trapezoid	✔️	✔️	✖

Support with JIT

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function	jax
cubature	✖
cumulative_simpson	✖
cumulative_trapezoid	✔️
dblquad	✖
fixed_quad	✖
lebedev_rule	✖
newton_cotes	✖
nquad	✖
nsum	✖
odeint	N/A
qmc_quad	✖
quad	✖
quad_vec	✖
romb	✔️
simpson	✖
solve_bvp	✖
solve_ivp	✖
tanhsinh	✖
tplquad	✖
trapezoid	✖