LICENSE
README.md
pyproject.toml
flowpde/__init__.py
flowpde/py.typed
flowpde.egg-info/PKG-INFO
flowpde.egg-info/SOURCES.txt
flowpde.egg-info/dependency_links.txt
flowpde.egg-info/requires.txt
flowpde.egg-info/top_level.txt
flowpde/core/__init__.py
flowpde/core/base_conditioner.py
flowpde/core/base_flow.py
flowpde/core/base_solver.py
flowpde/datasets/__init__.py
flowpde/datasets/normalization.py
flowpde/datasets/exponax/__init__.py
flowpde/datasets/exponax/base.py
flowpde/datasets/exponax/burgers.py
flowpde/datasets/exponax/darcy.py
flowpde/datasets/exponax/generator.py
flowpde/datasets/exponax/poisson.py
flowpde/datasets/exponax/utilities.py
flowpde/flows/__init__.py
flowpde/flows/neural_ode.py
flowpde/flows/components/__init__.py
flowpde/flows/components/couplings.py
flowpde/flows/components/paths.py
flowpde/flows/components/sources.py
flowpde/flows/components/time_samplers.py
flowpde/models/__init__.py
flowpde/models/components.py
flowpde/models/convnet.py
flowpde/models/mlp.py
flowpde/models/resnet.py
flowpde/models/unet.py
flowpde/objectives/__init__.py
flowpde/objectives/flow_matching.py
flowpde/objectives/maximum_likelihood.py
flowpde/solvers/__init__.py
flowpde/solvers/ode_solvers.py
flowpde/trainers/__init__.py
flowpde/trainers/ema.py
flowpde/trainers/evaluation.py
flowpde/trainers/reflow.py
flowpde/trainers/trainer.py
flowpde/utils/__init__.py
flowpde/utils/metrics.py
flowpde/utils/uq_metrics.py
flowpde/utils/utils.py
tests/test_components.py
tests/test_datasets_integration.py
tests/test_ema.py
tests/test_evaluation.py
tests/test_models_and_solver.py
tests/test_normalization.py
tests/test_reflow.py
tests/test_straightness.py
tests/test_trainer.py
tests/test_uq_metrics.py