gensbi.flow_matching.solver.solver#

Classes#

Abstract base class for solvers.

Module Contents#

class gensbi.flow_matching.solver.solver.Solver[source]#

Bases: abc.ABC

Abstract base class for solvers.

abstract sample(x_0)[source]#

Parameters:: x_0 (jax.Array)
Return type:: jax.Array