.gitignore
.pre-commit-config.yaml
AGENTS.md
CLAUDE.md
CODE_OF_CONDUCT.md
LICENSE
METHODS.md
README.md
RELEASES.rst
TESTING.md
codecov.yml
pyproject.toml
pytest.ini
requirements-ci.txt
setup.cfg
.cursor/rules
.github/CODEOWNERS
.github/PULL_REQUEST_TEMPLATE.md
.github/copilot-instructions.md
.github/workflows/docs.yml
.github/workflows/pre-commit.yml
.github/workflows/release.yaml
.github/workflows/testing.yml
assets/static_hf.json
assets/static_hf_parameters.json
assets/static_timm.json
assets/static_timm_parameters.json
assets/cli/plot_metric.py
assets/cli/summarize_csvlogs.py
assets/cm-unicode-0.7.0 2/Changes
assets/cm-unicode-0.7.0 2/FAQ
assets/cm-unicode-0.7.0 2/FontLog.txt
assets/cm-unicode-0.7.0 2/Fontmap.CMU
assets/cm-unicode-0.7.0 2/INSTALL
assets/cm-unicode-0.7.0 2/OFL-FAQ.txt
assets/cm-unicode-0.7.0 2/OFL.txt
assets/cm-unicode-0.7.0 2/README
assets/cm-unicode-0.7.0 2/TODO
assets/cm-unicode-0.7.0 2/cmunbbx.ttf
assets/cm-unicode-0.7.0 2/cmunbi.ttf
assets/cm-unicode-0.7.0 2/cmunbl.ttf
assets/cm-unicode-0.7.0 2/cmunbmo.ttf
assets/cm-unicode-0.7.0 2/cmunbmr.ttf
assets/cm-unicode-0.7.0 2/cmunbso.ttf
assets/cm-unicode-0.7.0 2/cmunbsr.ttf
assets/cm-unicode-0.7.0 2/cmunbtl.ttf
assets/cm-unicode-0.7.0 2/cmunbto.ttf
assets/cm-unicode-0.7.0 2/cmunbx.ttf
assets/cm-unicode-0.7.0 2/cmunbxo.ttf
assets/cm-unicode-0.7.0 2/cmunci.ttf
assets/cm-unicode-0.7.0 2/cmunit.ttf
assets/cm-unicode-0.7.0 2/cmunobi.ttf
assets/cm-unicode-0.7.0 2/cmunobx.ttf
assets/cm-unicode-0.7.0 2/cmunorm.ttf
assets/cm-unicode-0.7.0 2/cmunoti.ttf
assets/cm-unicode-0.7.0 2/cmunrb.ttf
assets/cm-unicode-0.7.0 2/cmunrm.ttf
assets/cm-unicode-0.7.0 2/cmunsi.ttf
assets/cm-unicode-0.7.0 2/cmunsl.ttf
assets/cm-unicode-0.7.0 2/cmunso.ttf
assets/cm-unicode-0.7.0 2/cmunss.ttf
assets/cm-unicode-0.7.0 2/cmunssdc.ttf
assets/cm-unicode-0.7.0 2/cmunst.ttf
assets/cm-unicode-0.7.0 2/cmunsx.ttf
assets/cm-unicode-0.7.0 2/cmuntb.ttf
assets/cm-unicode-0.7.0 2/cmunti.ttf
assets/cm-unicode-0.7.0 2/cmuntt.ttf
assets/cm-unicode-0.7.0 2/cmuntx.ttf
assets/cm-unicode-0.7.0 2/cmunui.ttf
assets/cm-unicode-0.7.0 2/cmunvi.ttf
assets/cm-unicode-0.7.0 2/cmunvt.ttf
assets/cm-unicode-0.7.0 2/fonts.scale
benchmarks/README.md
benchmarks/benchmark_masked_timm_vit.py
benchmarks/dataloaders.py
benchmarks/utils.py
benchmarks/cifar10/barlow-resnet18.py
benchmarks/cifar10/byol-resnet18.py
benchmarks/cifar10/nnclr-resnet18.py
benchmarks/cifar10/simclr-resnet18.py
benchmarks/cifar10/swav-resnet18.py
benchmarks/cifar10/vicreg-resnet18.py
benchmarks/data_loader/README.md
benchmarks/data_loader/benchmark_data_loading.py
benchmarks/diffusiondb2m/clip-vitb.py
benchmarks/imagenet10/RESULTS.md
benchmarks/imagenet10/barlow-vit-small.py
benchmarks/imagenet10/beit-vit-small.py
benchmarks/imagenet10/byol-vit-small.py
benchmarks/imagenet10/cmae-vit-small.py
benchmarks/imagenet10/collect_results.py
benchmarks/imagenet10/data2vec-vit-small.py
benchmarks/imagenet10/dino-vit-small.py
benchmarks/imagenet10/dinov2-vit-small.py
benchmarks/imagenet10/dinov3-vit-small.py
benchmarks/imagenet10/ibot-vit-small.py
benchmarks/imagenet10/igpt-vit-small.py
benchmarks/imagenet10/ijepa-vit-base.py
benchmarks/imagenet10/ijepa-vit-small.py
benchmarks/imagenet10/lejepa-vit-small-quick.py
benchmarks/imagenet10/lejepa-vit-small.py
benchmarks/imagenet10/mae-vit-base.py
benchmarks/imagenet10/mae-vit-small.py
benchmarks/imagenet10/masked.py
benchmarks/imagenet10/maskfeat-vit-small.py
benchmarks/imagenet10/mocov2-vit-small.py
benchmarks/imagenet10/mocov3-vit-small.py
benchmarks/imagenet10/msn-vit-small.py
benchmarks/imagenet10/multicrop.py
benchmarks/imagenet10/nnclr-vit-small.py
benchmarks/imagenet10/pirl-vit-small.py
benchmarks/imagenet10/simclr-vit-small.py
benchmarks/imagenet10/simmim-vit-small.py
benchmarks/imagenet10/simsiam-vit-small.py
benchmarks/imagenet10/swav-vit-small.py
benchmarks/imagenet10/tico-vit-small.py
benchmarks/imagenet10/two_view.py
benchmarks/imagenet10/vicreg-vit-small.py
benchmarks/imagenet10/vicregl-vit-small.py
benchmarks/imagenet10/wmse-vit-small.py
benchmarks/imagenet100/barlow-resnet50.py
benchmarks/imagenet100/byol-resnet50.py
benchmarks/imagenet100/clip-vitb-finetune.py
benchmarks/imagenet100/dino-resnet18.py
benchmarks/imagenet100/dino-vit-tiny.py
benchmarks/imagenet100/dinov2-vit-tiny-8gpus.py
benchmarks/imagenet100/dinov2-vit-tiny.py
benchmarks/imagenet100/ijepa-vit-base.py
benchmarks/imagenet100/mae-vit-base.py
benchmarks/imagenet100/nnclr-resnet50.py
benchmarks/imagenet100/salt-stage1-vit-tiny.py
benchmarks/imagenet100/salt-stage2-vit-tiny.py
benchmarks/imagenet100/simclr-resnet50.py
benchmarks/imagenet100/swav-resnet50.py
benchmarks/imagenet100/vicreg-resnet50.py
docs/.DS_Store
docs/Makefile
docs/make.bat
docs/source/bibliography.rst
docs/source/cache_dir.rst
docs/source/cli.rst
docs/source/conf.py
docs/source/contributing.rst
docs/source/index.rst
docs/source/references.bib
docs/source/releases.rst
docs/source/sg_execution_times.rst
docs/source/_templates/myclass_template.rst
docs/source/_templates/myfunc_template.rst
docs/source/api/backbone.rst
docs/source/api/data.rst
docs/source/api/forward.rst
docs/source/api/losses.rst
docs/source/api/manager.rst
docs/source/api/methods.rst
docs/source/api/module.rst
docs/source/api/monitors.rst
docs/source/api/optim.rst
docs/source/api/registry.rst
docs/source/api/web.rst
docs/source/figures/logo.jpg
examples/GALLERY_HEADER.rst
examples/hp_search.yaml
examples/imbalance_supervised_learning.py
examples/multi_layer_probe.py
examples/multi_probe.yaml
examples/simclr_cifar10_config.yaml
examples/simclr_cifar10_slurm.yaml
examples/simclr_cifar10_tutorial.ipynb
examples/supervised_learning.py
examples/wandb_figures.py
examples/wandb_reader.py
stable_pretraining/__about__.py
stable_pretraining/__init__.py
stable_pretraining/__init__.pyi
stable_pretraining/_config.py
stable_pretraining/_version.py
stable_pretraining/cli.py
stable_pretraining/config.py
stable_pretraining/forward.py
stable_pretraining/manager.py
stable_pretraining/module.py
stable_pretraining/py.typed
stable_pretraining/run.py
stable_pretraining/static.py
stable_pretraining.egg-info/PKG-INFO
stable_pretraining.egg-info/SOURCES.txt
stable_pretraining.egg-info/dependency_links.txt
stable_pretraining.egg-info/entry_points.txt
stable_pretraining.egg-info/requires.txt
stable_pretraining.egg-info/top_level.txt
stable_pretraining/backbone/__init__.py
stable_pretraining/backbone/aggregator.py
stable_pretraining/backbone/convmixer.py
stable_pretraining/backbone/decoders.py
stable_pretraining/backbone/mlp.py
stable_pretraining/backbone/nn_modules.py
stable_pretraining/backbone/patch_masking.py
stable_pretraining/backbone/pos_embed.py
stable_pretraining/backbone/probe.py
stable_pretraining/backbone/resnet9.py
stable_pretraining/backbone/utils.py
stable_pretraining/backbone/vit.py
stable_pretraining/backbone/video/__init__.py
stable_pretraining/backbone/video/causal_conv3d.py
stable_pretraining/backbone/video/cosmos.py
stable_pretraining/backbone/video/info.py
stable_pretraining/backbone/video/magvit2.py
stable_pretraining/backbone/video/norms.py
stable_pretraining/backbone/video/predrnn.py
stable_pretraining/backbone/video/recurrent_vit.py
stable_pretraining/backbone/video/videomamba.py
stable_pretraining/callbacks/README.md
stable_pretraining/callbacks/__init__.py
stable_pretraining/callbacks/checkpoint_sklearn.py
stable_pretraining/callbacks/checkpoint_swanlab.py
stable_pretraining/callbacks/checkpoint_trackio.py
stable_pretraining/callbacks/cleanup.py
stable_pretraining/callbacks/clip_zero_shot.py
stable_pretraining/callbacks/earlystop.py
stable_pretraining/callbacks/embedding_cache.py
stable_pretraining/callbacks/env_info.py
stable_pretraining/callbacks/factories.py
stable_pretraining/callbacks/hardware_monitor.py
stable_pretraining/callbacks/hf_models.py
stable_pretraining/callbacks/hp_metric.py
stable_pretraining/callbacks/image_decoder.py
stable_pretraining/callbacks/image_retrieval.py
stable_pretraining/callbacks/knn.py
stable_pretraining/callbacks/latent_viz.py
stable_pretraining/callbacks/lidar.py
stable_pretraining/callbacks/probe.py
stable_pretraining/callbacks/queue.py
stable_pretraining/callbacks/queues.py
stable_pretraining/callbacks/rankme.py
stable_pretraining/callbacks/registry.py
stable_pretraining/callbacks/teacher_student.py
stable_pretraining/callbacks/trainer_info.py
stable_pretraining/callbacks/unused_parameters.py
stable_pretraining/callbacks/utils.py
stable_pretraining/callbacks/wd_schedule.py
stable_pretraining/callbacks/writer.py
stable_pretraining/data/__init__.py
stable_pretraining/data/collate.py
stable_pretraining/data/dataset_stats.py
stable_pretraining/data/datasets.py
stable_pretraining/data/download.py
stable_pretraining/data/masking.py
stable_pretraining/data/module.py
stable_pretraining/data/sampler.py
stable_pretraining/data/synthetic_data.py
stable_pretraining/data/transforms.py
stable_pretraining/data/utils.py
stable_pretraining/data/video.py
stable_pretraining/loggers/__init__.py
stable_pretraining/loggers/csv_log_reader.py
stable_pretraining/loggers/swanlab.py
stable_pretraining/loggers/trackio.py
stable_pretraining/losses/__init__.py
stable_pretraining/losses/dino.py
stable_pretraining/losses/joint_embedding.py
stable_pretraining/losses/multimodal.py
stable_pretraining/losses/reconstruction.py
stable_pretraining/losses/utils.py
stable_pretraining/methods/__init__.py
stable_pretraining/methods/barlow_twins.py
stable_pretraining/methods/beit.py
stable_pretraining/methods/byol.py
stable_pretraining/methods/cmae.py
stable_pretraining/methods/data2vec.py
stable_pretraining/methods/dino.py
stable_pretraining/methods/dinov2.py
stable_pretraining/methods/dinov3.py
stable_pretraining/methods/ibot.py
stable_pretraining/methods/igpt.py
stable_pretraining/methods/ijepa.py
stable_pretraining/methods/lejepa.py
stable_pretraining/methods/mae.py
stable_pretraining/methods/maskfeat.py
stable_pretraining/methods/mim_refiner.py
stable_pretraining/methods/mocov2.py
stable_pretraining/methods/mocov3.py
stable_pretraining/methods/msn.py
stable_pretraining/methods/nepa.py
stable_pretraining/methods/nnclr.py
stable_pretraining/methods/pirl.py
stable_pretraining/methods/salt.py
stable_pretraining/methods/simclr.py
stable_pretraining/methods/simmim.py
stable_pretraining/methods/simsiam.py
stable_pretraining/methods/swav.py
stable_pretraining/methods/tico.py
stable_pretraining/methods/vicreg.py
stable_pretraining/methods/vicregl.py
stable_pretraining/methods/wmse.py
stable_pretraining/optim/__init__.py
stable_pretraining/optim/lars.py
stable_pretraining/optim/lr_scheduler.py
stable_pretraining/optim/utils.py
stable_pretraining/registry/__init__.py
stable_pretraining/registry/_scanner.py
stable_pretraining/registry/_sidecar.py
stable_pretraining/registry/_store.py
stable_pretraining/registry/logger.py
stable_pretraining/registry/query.py
stable_pretraining/tests/README.md
stable_pretraining/tests/conftest.py
stable_pretraining/tests/utils.py
stable_pretraining/tests/integration/__init__.py
stable_pretraining/tests/integration/test_backbone.py
stable_pretraining/tests/integration/test_datasets.py
stable_pretraining/tests/integration/test_ddp_requeue.py
stable_pretraining/tests/integration/test_ijepa_inet10.py
stable_pretraining/tests/integration/test_ijepa_module_input.py
stable_pretraining/tests/integration/test_image_retrieval.py
stable_pretraining/tests/integration/test_lejepa_inet10.py
stable_pretraining/tests/integration/test_mae.py
stable_pretraining/tests/integration/test_mae_inet10.py
stable_pretraining/tests/integration/test_mae_module_input.py
stable_pretraining/tests/integration/test_mnist.py
stable_pretraining/tests/integration/test_probing.py
stable_pretraining/tests/integration/test_simclr.py
stable_pretraining/tests/integration/test_supervised.py
stable_pretraining/tests/integration/test_video_loading.py
stable_pretraining/tests/integration/test_writer.py
stable_pretraining/tests/regression/__init__.py
stable_pretraining/tests/regression/_capture_refs.py
stable_pretraining/tests/regression/test_methods.py
stable_pretraining/tests/scripts/regtest_single_multi_gpu_fast.py
stable_pretraining/tests/scripts/regtest_single_multi_gpu_slow.py
stable_pretraining/tests/scripts/test_wandb_callback.py
stable_pretraining/tests/scripts/train_simclr_cifar10_multigpu.py
stable_pretraining/tests/scripts/train_simclr_cifar10_multiopt.py
stable_pretraining/tests/scripts/train_supervised_multigpu_requeue.py
stable_pretraining/tests/unit/__init__.py
stable_pretraining/tests/unit/test_aggregator.py
stable_pretraining/tests/unit/test_atomic_checkpoint.py
stable_pretraining/tests/unit/test_autotune_mlp.py
stable_pretraining/tests/unit/test_backbone.py
stable_pretraining/tests/unit/test_best_compressed.py
stable_pretraining/tests/unit/test_cache_dir.py
stable_pretraining/tests/unit/test_callback_ordering_log.py
stable_pretraining/tests/unit/test_callback_utils.py
stable_pretraining/tests/unit/test_callback_verbose_logging.py
stable_pretraining/tests/unit/test_checkpointing.py
stable_pretraining/tests/unit/test_cleanup.py
stable_pretraining/tests/unit/test_clip_loss.py
stable_pretraining/tests/unit/test_clip_zero_shot.py
stable_pretraining/tests/unit/test_cosmos.py
stable_pretraining/tests/unit/test_csvlog_reader.py
stable_pretraining/tests/unit/test_datasets.py
stable_pretraining/tests/unit/test_detach_tensors.py
stable_pretraining/tests/unit/test_env_callback.py
stable_pretraining/tests/unit/test_error_handling.py
stable_pretraining/tests/unit/test_flops.py
stable_pretraining/tests/unit/test_forward_functions.py
stable_pretraining/tests/unit/test_global_config.py
stable_pretraining/tests/unit/test_global_exclude_bias_norm.py
stable_pretraining/tests/unit/test_hardware_monitor.py
stable_pretraining/tests/unit/test_hf_models.py
stable_pretraining/tests/unit/test_ibot_patch_loss.py
stable_pretraining/tests/unit/test_ijepa_masking.py
stable_pretraining/tests/unit/test_image_decoder.py
stable_pretraining/tests/unit/test_image_retrieval.py
stable_pretraining/tests/unit/test_image_retrieval_batched.py
stable_pretraining/tests/unit/test_knn_callback.py
stable_pretraining/tests/unit/test_load_callback.py
stable_pretraining/tests/unit/test_mae.py
stable_pretraining/tests/unit/test_mae_decoder.py
stable_pretraining/tests/unit/test_magvit2.py
stable_pretraining/tests/unit/test_manager.py
stable_pretraining/tests/unit/test_masked_encoder.py
stable_pretraining/tests/unit/test_masking.py
stable_pretraining/tests/unit/test_methods_smoke.py
stable_pretraining/tests/unit/test_metrics.py
stable_pretraining/tests/unit/test_mnist.py
stable_pretraining/tests/unit/test_module.py
stable_pretraining/tests/unit/test_multi_block_masking.py
stable_pretraining/tests/unit/test_nn_modules.py
stable_pretraining/tests/unit/test_nnclr.py
stable_pretraining/tests/unit/test_optim_utils.py
stable_pretraining/tests/unit/test_patch_masking.py
stable_pretraining/tests/unit/test_pos_embed.py
stable_pretraining/tests/unit/test_predrnn.py
stable_pretraining/tests/unit/test_probing.py
stable_pretraining/tests/unit/test_queue_callback.py
stable_pretraining/tests/unit/test_queue_state_dict.py
stable_pretraining/tests/unit/test_recurrent_vit.py
stable_pretraining/tests/unit/test_registry.py
stable_pretraining/tests/unit/test_run_registry.py
stable_pretraining/tests/unit/test_salt.py
stable_pretraining/tests/unit/test_samplers.py
stable_pretraining/tests/unit/test_simclr.py
stable_pretraining/tests/unit/test_slurm_index.py
stable_pretraining/tests/unit/test_solver.py
stable_pretraining/tests/unit/test_streamingtopk.py
stable_pretraining/tests/unit/test_supervised.py
stable_pretraining/tests/unit/test_swanlab.py
stable_pretraining/tests/unit/test_timm_vit_patch.py
stable_pretraining/tests/unit/test_trackio.py
stable_pretraining/tests/unit/test_transforms.py
stable_pretraining/tests/unit/test_unused_params.py
stable_pretraining/tests/unit/test_video.py
stable_pretraining/tests/unit/test_video_loading.py
stable_pretraining/tests/unit/test_video_transforms.py
stable_pretraining/tests/unit/test_videomamba.py
stable_pretraining/tests/unit/test_vit.py
stable_pretraining/tests/unit/test_vit_classifier.py
stable_pretraining/tests/unit/test_writer.py
stable_pretraining/utils/__init__.py
stable_pretraining/utils/atomic_checkpoint.py
stable_pretraining/utils/config.py
stable_pretraining/utils/data_generation.py
stable_pretraining/utils/distance_metrics.py
stable_pretraining/utils/distributed.py
stable_pretraining/utils/error_handling.py
stable_pretraining/utils/flops.py
stable_pretraining/utils/gdrive_utils.py
stable_pretraining/utils/inspection_utils.py
stable_pretraining/utils/lightning_patch.py
stable_pretraining/utils/log_reader.py
stable_pretraining/utils/online_topk.py
stable_pretraining/utils/solver.py
stable_pretraining/utils/stats.py
stable_pretraining/utils/timm_to_hf_hub.py
stable_pretraining/utils/visualization.py
stable_pretraining/web/__init__.py
stable_pretraining/web/scan.py
stable_pretraining/web/server.py
stable_pretraining/web/assets/app.css
stable_pretraining/web/assets/app.js
stable_pretraining/web/assets/index.html