LICENSE
LICENSE.txt
README.md
pyproject.toml
pdetransformer/__init__.py
pdetransformer/setup.py
pdetransformer/utils.py
pdetransformer/visualization.py
pdetransformer.egg-info/PKG-INFO
pdetransformer.egg-info/SOURCES.txt
pdetransformer.egg-info/dependency_links.txt
pdetransformer.egg-info/requires.txt
pdetransformer.egg-info/top_level.txt
pdetransformer/callback/__init__.py
pdetransformer/callback/callbacks.py
pdetransformer/callback/diffusers.py
pdetransformer/callback/ema.py
pdetransformer/callback/ema_clip.py
pdetransformer/callback/setup_callback.py
pdetransformer/callback/simulation_2d_metrics.py
pdetransformer/callback/videos.py
pdetransformer/core/__init__.py
pdetransformer/core/mixed_channels/__init__.py
pdetransformer/core/mixed_channels/dit.py
pdetransformer/core/mixed_channels/factformer.py
pdetransformer/core/mixed_channels/pde_transformer.py
pdetransformer/core/mixed_channels/scot.py
pdetransformer/core/mixed_channels/train_probabilistic.py
pdetransformer/core/mixed_channels/train_supervised.py
pdetransformer/core/mixed_channels/udit.py
pdetransformer/core/mixed_channels/unet.py
pdetransformer/core/separate_channels/__init__.py
pdetransformer/core/separate_channels/pde_transformer.py
pdetransformer/core/separate_channels/tasks.py
pdetransformer/core/separate_channels/train_probabilistic.py
pdetransformer/core/separate_channels/train_supervised.py
pdetransformer/core/separate_channels/utils.py
pdetransformer/data/__init__.py
pdetransformer/data/cached_dataset.py
pdetransformer/data/joint_module.py
pdetransformer/data/metadata_dataset.py
pdetransformer/data/metadata_remapping.py
pdetransformer/data/multi_module.py
pdetransformer/data/pbdl_module.py
pdetransformer/data/script_convert_torchfsm.py
pdetransformer/data/utils.py
pdetransformer/data/pbdl_dataloader/__init__.py
pdetransformer/data/pbdl_dataloader/colors.py
pdetransformer/data/pbdl_dataloader/dataset.py
pdetransformer/data/pbdl_dataloader/fetcher.py
pdetransformer/data/pbdl_dataloader/loader.py
pdetransformer/data/pbdl_dataloader/logging.py
pdetransformer/data/pbdl_dataloader/normalization.py
pdetransformer/data/pbdl_dataloader/utilities.py
pdetransformer/data/pbdl_dataloader/torch/__init__.py
pdetransformer/data/pbdl_dataloader/torch/dataset.py
pdetransformer/data/pbdl_dataloader/torch/loader.py
pdetransformer/data/pbdl_datatypes/__init__.py
pdetransformer/data/pbdl_datatypes/acdm_2d.py
pdetransformer/data/pbdl_datatypes/ape_2d.py
pdetransformer/data/pbdl_datatypes/ape_2d_xxl.py
pdetransformer/data/pbdl_datatypes/variable_dt_dataset.py
pdetransformer/data/pbdl_datatypes/well_2d.py
pdetransformer/metric/__init__.py
pdetransformer/metric/metric.py
pdetransformer/metric/mse.py
pdetransformer/objectives/__init__.py
pdetransformer/objectives/cot_flow_matching.py
pdetransformer/objectives/diffusion_markov_chain.py
pdetransformer/objectives/diffusion_markov_chain_coupled.py
pdetransformer/objectives/lipman_flow_matching.py
pdetransformer/objectives/losses.py
pdetransformer/objectives/training_objective.py
pdetransformer/objectives/utils.py
pdetransformer/objectives/diffusion/__init__.py
pdetransformer/objectives/diffusion/diffusion_base.py
pdetransformer/objectives/diffusion/markov_chain.py
pdetransformer/objectives/paths/__init__.py
pdetransformer/objectives/paths/conditional_optimal_transport.py
pdetransformer/objectives/paths/path_base.py
pdetransformer/objectives/paths/sde.py
pdetransformer/sampler/__init__.py
pdetransformer/sampler/ancestral_sampler.py
pdetransformer/sampler/sampler_base.py
pdetransformer/sampler/pipeline/__init__.py
pdetransformer/sampler/pipeline/coupled_video.py
pdetransformer/sampler/scheduler/__init__.py
pdetransformer/sampler/scheduler/ode_euler.py