Array API Standard Support: integrate

Support on CPU

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

function/class	torch	jax	dask
BDF	✖	✖	✖
DOP853	✖	✖	✖
DenseOutput	✖	✖	✖
LSODA	✖	✖	✖
OdeSolution	✖	✖	✖
OdeSolver	✖	✖	✖
RK23	✖	✖	✖
RK45	✖	✖	✖
Radau	✖	✖	✖
complex_ode	N/A	N/A	N/A
cubature	✔️	✔️	✔️
cumulative_simpson	✔️	✔️	✔️
cumulative_trapezoid	✔️	✔️	✔️
dblquad	✖	✖	✖
fixed_quad	✔️	✔️	✔️
lebedev_rule	✖	✖	✖
newton_cotes	✖	✖	✖
nquad	✖	✖	✖
nsum	✖	✖	✖
ode	N/A	N/A	N/A
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/class	cupy	torch	jax
BDF	✖	✖	✖
DOP853	✖	✖	✖
DenseOutput	✖	✖	✖
LSODA	✖	✖	✖
OdeSolution	✖	✖	✖
OdeSolver	✖	✖	✖
RK23	✖	✖	✖
RK45	✖	✖	✖
Radau	✖	✖	✖
complex_ode	N/A	N/A	N/A
cubature	✔️	✔️	✔️
cumulative_simpson	✔️	✔️	✔️
cumulative_trapezoid	✔️	✔️	✔️
dblquad	✖	✖	✖
fixed_quad	✔️	✔️	✔️
lebedev_rule	✖	✖	✖
newton_cotes	✖	✖	✖
nquad	✖	✖	✖
nsum	✔️	✖	✖
ode	N/A	N/A	N/A
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/class	jax
BDF	✖
DOP853	✖
DenseOutput	✖
LSODA	✖
OdeSolution	✖
OdeSolver	✖
RK23	✖
RK45	✖
Radau	✖
complex_ode	N/A
cubature	✖
cumulative_simpson	✖
cumulative_trapezoid	✔️
dblquad	✖
fixed_quad	✔️
lebedev_rule	✖
newton_cotes	✖
nquad	✖
nsum	✖
ode	N/A
odeint	N/A
qmc_quad	✖
quad	✖
quad_vec	✖
romb	✔️
simpson	✔️
solve_bvp	✖
solve_ivp	✖
tanhsinh	✖
tplquad	✖
trapezoid	✔️