Expand description
Minimal ODE integrators for sampling flow models.
This crate uses ODE sampling of the form: [ \frac{dx}{dt} = v_\theta(x,t;\cdot) ]
We keep this module tiny and deterministic: no adaptive stepping, no hidden tolerances.
§Discretization error analysis
Guan et al., “Total Variation Rates for Riemannian Flow Matching” (2026) provides the first nonasymptotic TV convergence analysis for Riemannian flow matching sampling. It quantifies how Euler discretization error propagates to sample quality, giving concrete bounds on the number of integration steps needed for a target TV tolerance.
Enums§
- OdeMethod
- Fixed-step ODE method.
Functions§
- integrate_
fixed - Integrate an ODE forward with fixed steps on a vector state.