LICENSE
README.md
setup.py
demo/__init__.py
demo/demo.py
diff_plonk.egg-info/PKG-INFO
diff_plonk.egg-info/SOURCES.txt
diff_plonk.egg-info/dependency_links.txt
diff_plonk.egg-info/requires.txt
diff_plonk.egg-info/top_level.txt
plonk/__init__.py
plonk/pipe.py
plonk/train.py
plonk/train_random.py
plonk/train_von_fisher.py
plonk/callbacks/__init__.py
plonk/callbacks/data.py
plonk/callbacks/ema.py
plonk/callbacks/fix_nans.py
plonk/data/__init__.py
plonk/data/augmentation.py
plonk/data/data.py
plonk/data/datamodule.py
plonk/data/transforms.py
plonk/data/webdataset.py
plonk/data/extract_embeddings/__init__.py
plonk/data/extract_embeddings/dataset_with_path.py
plonk/data/extract_embeddings/dino_v2.py
plonk/data/extract_embeddings/launch_embedding_extraction.py
plonk/data/extract_embeddings/so_siglip.py
plonk/data/extract_embeddings/street_clip.py
plonk/metrics/__init__.py
plonk/metrics/distance_based.py
plonk/metrics/elo.py
plonk/metrics/utils.py
plonk/models/__init__.py
plonk/models/losses.py
plonk/models/module.py
plonk/models/positional_embeddings.py
plonk/models/postprocessing.py
plonk/models/preconditioning.py
plonk/models/preprocessing.py
plonk/models/pretrained_models.py
plonk/models/schedulers.py
plonk/models/networks/__init__.py
plonk/models/networks/mlp.py
plonk/models/networks/transformers.py
plonk/models/samplers/__init__.py
plonk/models/samplers/ddim.py
plonk/models/samplers/ddpm.py
plonk/models/samplers/edm.py
plonk/models/samplers/flow_sampler.py
plonk/models/samplers/riemannian_flow_sampler.py
plonk/models/samplers/von_fisher_sampling.py
plonk/utils/__init__.py
plonk/utils/image_processing.py
plonk/utils/kde.py
plonk/utils/lr_scheduler.py
plonk/utils/manifolds.py
plonk/utils/model_utils.py
plonk/utils/optimizers.py