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

function/class

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

qmc_quad

✖

quad

✖

quad_vec

✖

romb

âœ”ī¸

simpson

âœ”ī¸

solve_bvp

✖

solve_ivp

✖

tanhsinh

✖

tplquad

✖

trapezoid

✖