nami.solvers.ode

Classes

RK4([steps])

class nami.solvers.ode.RK4(steps=32)

Bases: object

Parameters:

steps (int)

integrate(f, x0, *, t0, t1, atol=1e-06, rtol=1e-05, steps=None)

Return type:

Tensor

Parameters:

• x0 (Tensor)

• t0 (float)

• t1 (float)

• atol (float)

• rtol (float)

• steps (int | None)

integrate_augmented(f_aug, x0, logp0, *, t0, t1, atol=1e-06, rtol=1e-05, steps=None)

Return type:

tuple[Tensor, Tensor]

Parameters:

• x0 (Tensor)

• logp0 (Tensor)

• t0 (float)

• t1 (float)

• atol (float)

• rtol (float)

• steps (int | None)

is_sde = False

requires_steps = True

supports_rsample = True