Array API Standard Support: integrate#

Support on CPU#

Legend

✔️ = supported

✖ = unsupported

N/A = out-of-scope

blank = not currently documented

function/class

torch

jax

dask

BDF

DOP853

DenseOutput

LSODA

OdeSolution

OdeSolver

RK23

RK45

Radau

complex_ode

cubature

✔️

✔️

✔️

cumulative_simpson

✔️

✔️

cumulative_trapezoid

✔️

✔️

✔️

dblquad

fixed_quad

✔️

✔️

✔️

lebedev_rule

newton_cotes

nquad

nsum

ode

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

blank = not currently documented

function/class

cupy

torch

jax

BDF

DOP853

DenseOutput

LSODA

OdeSolution

OdeSolver

RK23

RK45

Radau

complex_ode

cubature

✔️

✔️

✔️

cumulative_simpson

✔️

✔️

cumulative_trapezoid

✔️

✔️

✔️

dblquad

fixed_quad

✔️

✔️

✔️

lebedev_rule

newton_cotes

nquad

nsum

✔️

ode

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

blank = not currently documented