All modules for which code is available

• flow_matching.loss.generalized_loss
• flow_matching.path.affine
• flow_matching.path.geodesic
• flow_matching.path.mixture
• flow_matching.path.path
• flow_matching.path.path_sample
• flow_matching.path.scheduler.schedule_transform
• flow_matching.path.scheduler.scheduler
• flow_matching.solver.discrete_solver
• flow_matching.solver.ode_solver
• flow_matching.solver.riemannian_ode_solver
• flow_matching.solver.solver
• flow_matching.utils.manifolds.manifold
• flow_matching.utils.manifolds.sphere
• flow_matching.utils.manifolds.torus
• flow_matching.utils.manifolds.utils
• flow_matching.utils.model_wrapper