LICENSE
README.md
setup.py
flow_matching/__init__.py
flow_matching.egg-info/PKG-INFO
flow_matching.egg-info/SOURCES.txt
flow_matching.egg-info/dependency_links.txt
flow_matching.egg-info/requires.txt
flow_matching.egg-info/top_level.txt
flow_matching/loss/__init__.py
flow_matching/loss/generalized_loss.py
flow_matching/path/__init__.py
flow_matching/path/affine.py
flow_matching/path/geodesic.py
flow_matching/path/mixture.py
flow_matching/path/path.py
flow_matching/path/path_sample.py
flow_matching/path/scheduler/__init__.py
flow_matching/path/scheduler/schedule_transform.py
flow_matching/path/scheduler/scheduler.py
flow_matching/solver/__init__.py
flow_matching/solver/discrete_solver.py
flow_matching/solver/ode_solver.py
flow_matching/solver/riemannian_ode_solver.py
flow_matching/solver/solver.py
flow_matching/solver/utils.py
flow_matching/utils/__init__.py
flow_matching/utils/categorical_sampler.py
flow_matching/utils/model_wrapper.py
flow_matching/utils/utils.py
flow_matching/utils/manifolds/__init__.py
flow_matching/utils/manifolds/manifold.py
flow_matching/utils/manifolds/sphere.py
flow_matching/utils/manifolds/torus.py
flow_matching/utils/manifolds/utils.py
tests/__init__.py
tests/path/__init__.py
tests/path/test_path.py
tests/path/test_schedule_transform.py
tests/path/test_scheduler.py
tests/solver/__init__.py
tests/solver/test_discrete_solver.py
tests/solver/test_ode_solver.py
tests/solver/test_riemannian_ode_solver.py
tests/utils/__init__.py
tests/utils/test_utils.py