.env.example
.gitignore
CLAUDE.md
INSTALLATION_AND_VERIFICATION.md
LICENSE
MIGRATION_GUIDE.md
PLAN_TRAINING_LIBRARY.md
QUICK_START.md
README.md
REORGANIZATION_COMPLETE.md
SHELL_SCRIPTS_FIXED.md
TODO_upload_hf.md
VAE_SUMMARY.md
VQVAE_IMPLEMENTATION.md
fix_installation.sh
generate_example.py
mkdocs.yml
pyproject.toml
requirements-docs.txt
upload_checkpoints_example.py
verify_reorganization.py
.claude/skills/visual-docs/SKILL.md
.claude/skills/visual-docs/scripts/screenshot_docs.py
.github/workflows/docs.yml
.github/workflows/publish.yml
configs/README.md
configs/upload_checkpoints.yaml
configs/dataset/mimic.yaml
configs/dataset/ptbxl.yaml
configs/experiments/cvae_mimic.yaml
configs/experiments/diffusion_mimic.yaml
configs/experiments/diffusion_mimic_conditional.yaml
configs/experiments/diffusion_mimic_leads.yaml
configs/experiments/diffusion_mimic_reports.yaml
configs/experiments/diffusion_mimic_ssm.yaml
configs/experiments/flow_matching_mimic.yaml
configs/experiments/latent_diffusion_mimic.yaml
configs/experiments/prior_mimic.yaml
configs/experiments/prior_mimic_conditional.yaml
configs/experiments/ptbxl_sssd_ecg.yaml
configs/experiments/ptbxl_text_demographics.yaml
configs/experiments/pulse2pulse_lead_reconstruction.yaml
configs/experiments/pulse2pulse_mimic.yaml
configs/experiments/vae_mimic.yaml
configs/experiments/vqvae_mimic.yaml
configs/model/imn.yaml
configs/model/gan/pulse2pulse.yaml
configs/model/vae/vae_1d_base.yaml
configs/model/vae/vae_1d_large.yaml
configs/model/vqvae/prior_base.yaml
configs/model/vqvae/vqvae_1d_base.yaml
configs/training/default.yaml
configs/training/fast_dev.yaml
configs/training/production.yaml
data/README.md
data/__init__.py
data/datasets/__init__.py
data/datasets/base.py
data/datasets/mimic/README.md
data/datasets/mimic/__init__.py
data/datasets/mimic/dataset.py
data/datasets/mimic/pulse2pulse_datamodule.py
docs/changelog.md
docs/concepts.md
docs/extending.md
docs/index.md
docs/project_structure.md
docs/reorganization_history.md
docs/api/conditioning.md
docs/api/data.md
docs/api/evaluation.md
docs/api/models.md
docs/api/training.md
docs/api/utils.md
docs/api/visualization.md
docs/assets/favicon.svg
docs/datasets/custom.md
docs/datasets/mimic.md
docs/datasets/ptbxl.md
docs/evaluation/metrics.md
docs/getting_started/configuration.md
docs/getting_started/installation.md
docs/getting_started/quickstart.md
docs/javascripts/architecture.js
docs/javascripts/mathjax.js
docs/models/architecture.md
docs/models/diffusion.md
docs/models/gan.md
docs/models/vae.md
docs/models/vqvae.md
docs/standalone/vqvae_standalone.md
docs/stylesheets/architecture.css
docs/training/callbacks.md
docs/training/conditioning.md
docs/training/guide.md
docs/training/pulse2pulse_training.md
docs/training/runner.md
docs/training/sampling.md
docs/training/scripts_comparison.md
docs/training/vae_training.md
docs/training/vqvae_training.md
docs/visual/diffusion.md
docs/visual/flow_matching.md
docs/visual/index.md
docs/visual/pulse2pulse.md
docs/visual/vae.md
docs/visual/vqvae.md
docs/visualization/guide.md
evaluation/README.md
evaluation/__init__.py
evaluation/benchmark.py
evaluation/visualize.py
evaluation/metrics/__init__.py
evaluation/metrics/diversity.py
evaluation/metrics/fidelity.py
evaluation/metrics/signal_quality.py
models/README.md
models/__init__.py
models/components/__init__.py
models/components/blocks.py
models/components/decoders.py
models/components/encoders.py
models/diffusion/README.md
models/diffusion/__init__.py
models/gan/README.md
models/gan/__init__.py
models/gan/pulse2pulse.py
models/ssm/README.md
models/ssm/__init__.py
models/vae/README.md
models/vae/__init__.py
models/vae/config.py
models/vae/vae_1d.py
models/vae/vae_lightning.py
models/vqvae/README.md
models/vqvae/__init__.py
models/vqvae/config.py
models/vqvae/pixelcnn_prior.py
models/vqvae/prior_lightning.py
models/vqvae/vector_quantizer.py
models/vqvae/vqvae_1d.py
models/vqvae/vqvae_lightning.py
runs/pulse2pulse_mimic/seed_42/config_resolved.yaml
runs/pulse2pulse_mimic/seed_42/metadata.json
runs/pulse2pulse_mimic/seed_42/tb/version_0/events.out.tfevents.1772121525.mimir.3649231.0
runs/pulse2pulse_mimic/seed_42/tb/version_0/hparams.yaml
scripts/README.md
scripts/README_VAE_MIMIC.md
scripts/README_VAE_TRAINING.md
scripts/README_VISUALIZATION.md
scripts/upload_checkpoints_to_hf.py
scripts/data/prepare_mimic.py
scripts/data/validate_dataset.py
scripts/evaluate/benchmark_models.py
scripts/evaluate/condition_adherence.py
scripts/evaluate/evaluate_model.py
scripts/evaluate/lead_reconstruction.py
scripts/generate/generate_pulse2pulse.py
scripts/generate/generate_vae.py
scripts/shell/run_sweep.sh
scripts/shell/run_train_pulse2pulse_mimic.sh
scripts/shell/run_train_pulse2pulse_standalone.sh
scripts/shell/run_train_vae_mimic_config.sh
scripts/shell/run_train_vqvae_mimic.sh
scripts/shell/run_train_vqvae_prior_mimic.sh
scripts/train/train_template.py
scripts/utils/test_models_only.py
scripts/utils/test_pulse2pulse_setup.py
single_file/FIXES.md
single_file/QUICK_START.md
single_file/README.md
single_file/SUMMARY.md
single_file/WANDB_GUIDE.md
single_file/run_train_vqvae.sh
single_file/test_data_loading.py
single_file/train_vqvae_standalone.py
src/ecgen/__init__.py
src/ecgen/__main__.py
src/ecgen/_version.py
src/ecgen/api.py
src/ecgen/cli.py
src/ecgen/generate.py
src/ecgen/plot.py
src/ecgen/registry.py
src/ecgen/types.py
src/ecgen.egg-info/PKG-INFO
src/ecgen.egg-info/SOURCES.txt
src/ecgen.egg-info/dependency_links.txt
src/ecgen.egg-info/entry_points.txt
src/ecgen.egg-info/requires.txt
src/ecgen.egg-info/scm_file_list.json
src/ecgen.egg-info/scm_version.json
src/ecgen.egg-info/top_level.txt
src/ecgen/conditioning/__init__.py
src/ecgen/conditioning/composite.py
src/ecgen/conditioning/encoders.py
src/ecgen/conditioning/text.py
src/ecgen/data/__init__.py
src/ecgen/data/collate.py
src/ecgen/data/datamodule.py
src/ecgen/data/leads.py
src/ecgen/data/masks.py
src/ecgen/data/mimic_dataset.py
src/ecgen/data/normalizers.py
src/ecgen/data/pulse2pulse_mimic.py
src/ecgen/data/datasets/__init__.py
src/ecgen/data/datasets/base.py
src/ecgen/data/datasets/mimic.py
src/ecgen/data/datasets/ptbxl.py
src/ecgen/data/datasets/synthetic.py
src/ecgen/evaluation/__init__.py
src/ecgen/evaluation/classifier.py
src/ecgen/evaluation/features.py
src/ecgen/evaluation/measure.py
src/ecgen/evaluation/metrics.py
src/ecgen/f/__init__.py
src/ecgen/f/upload_hf.py
src/ecgen/families/__init__.py
src/ecgen/families/base.py
src/ecgen/families/continuous/__init__.py
src/ecgen/families/continuous/latent.py
src/ecgen/families/continuous/module.py
src/ecgen/families/gan/__init__.py
src/ecgen/families/gan/losses.py
src/ecgen/families/gan/pulse2pulse.py
src/ecgen/families/vae/__init__.py
src/ecgen/families/vae/config.py
src/ecgen/families/vae/losses.py
src/ecgen/families/vae/vae_1d.py
src/ecgen/families/vae/vae_lightning.py
src/ecgen/families/vqvae/__init__.py
src/ecgen/families/vqvae/config.py
src/ecgen/families/vqvae/losses.py
src/ecgen/families/vqvae/pixelcnn_prior.py
src/ecgen/families/vqvae/prior_lightning.py
src/ecgen/families/vqvae/vector_quantizer.py
src/ecgen/families/vqvae/vqvae_1d.py
src/ecgen/families/vqvae/vqvae_lightning.py
src/ecgen/models/__init__.py
src/ecgen/models/pulse2pulse.py
src/ecgen/models/vae.py
src/ecgen/nn/__init__.py
src/ecgen/nn/blocks.py
src/ecgen/nn/decoders.py
src/ecgen/nn/embeddings.py
src/ecgen/nn/encoders.py
src/ecgen/nn/injection.py
src/ecgen/nn/backbones/__init__.py
src/ecgen/nn/backbones/dit1d.py
src/ecgen/nn/backbones/ssm.py
src/ecgen/nn/backbones/unet1d.py
src/ecgen/processes/__init__.py
src/ecgen/processes/base.py
src/ecgen/processes/ddpm.py
src/ecgen/processes/flow.py
src/ecgen/training/__init__.py
src/ecgen/training/ema.py
src/ecgen/training/losses.py
src/ecgen/training/metrics.py
src/ecgen/training/runner.py
src/ecgen/training/test.py
src/ecgen/training/train.py
src/ecgen/training/validate.py
src/ecgen/training/callbacks/__init__.py
src/ecgen/training/callbacks/checkpointing.py
src/ecgen/training/callbacks/logging.py
src/ecgen/training/callbacks/visualization.py
src/ecgen/utils/__init__.py
src/ecgen/utils/config.py
src/ecgen/utils/io.py
src/ecgen/utils/logging.py
src/ecgen/utils/metadata.py
src/ecgen/utils/seed.py
tests/__init__.py
tests/test_upload_hf.py
tests/test_conditioning/__init__.py
tests/test_conditioning/test_conditioning.py
tests/test_conditioning/test_text_and_ptbxl.py
tests/test_conditioning/toy.py
tests/test_core/__init__.py
tests/test_core/test_registry.py
tests/test_core/test_shims.py
tests/test_data/__init__.py
tests/test_data/test_ecgen_data.py
tests/test_data/test_mimic_dataset.py
tests/test_evaluation/__init__.py
tests/test_evaluation/test_evaluation_and_api.py
tests/test_models/__init__.py
tests/test_models/test_checkpoint_loading.py
tests/test_models/test_conditional_families.py
tests/test_models/test_diffusion.py
tests/test_models/test_flow_matching.py
tests/test_models/test_latent_and_backbones.py
tests/test_models/test_pixelcnn_prior.py
tests/test_models/test_pulse2pulse.py
tests/test_models/test_vae.py
tests/test_models/test_vqvae_sampling.py
tests/test_training/__init__.py
tests/test_training/test_runner.py
training/README.md
training/__init__.py
training/callbacks/__init__.py
training/callbacks/checkpointing.py
training/callbacks/logging.py
training/callbacks/visualization.py
training/losses/__init__.py
training/losses/common.py
training/losses/gan_losses.py
training/losses/vae_losses.py
training/losses/vqvae_losses.py
training/metrics/__init__.py
training/metrics/clinical.py
training/metrics/generation.py
training/metrics/reconstruction.py
training/trainers/__init__.py
training/trainers/base_trainer.py
training/trainers/gan_trainer.py
training/trainers/vae_trainer.py
utils/__init__.py
utils/config.py
utils/io.py
utils/logging.py
utils/metadata.py
utils/seed.py
visualization/README.md
visualization/__init__.py
visualization/ecg_plots.py
visualization/latent_space.py
visualization/training_curves.py