LICENSE
README.md
setup.py
src/algorithms/__init__.py
src/algorithms/core_algorithm.py
src/algorithms/inference.py
src/algorithms/optimization.py
src/algorithms/sampling.py
src/data/__init__.py
src/data/collate.py
src/data/datasets.py
src/data/download_prepare.py
src/data/transforms.py
src/eval/__init__.py
src/eval/evaluators.py
src/eval/figures.py
src/eval/metrics.py
src/eval/tables.py
src/losses/__init__.py
src/losses/auxiliary_losses.py
src/losses/main_losses.py
src/losses/metrics_loss_utils.py
src/models/__init__.py
src/models/backbone.py
src/models/embeddings.py
src/models/heads.py
src/models/initializers.py
src/models/modules.py
src/models/ssa_transformer.py
src/selective_self_attention.egg-info/PKG-INFO
src/selective_self_attention.egg-info/SOURCES.txt
src/selective_self_attention.egg-info/dependency_links.txt
src/selective_self_attention.egg-info/entry_points.txt
src/selective_self_attention.egg-info/not-zip-safe
src/selective_self_attention.egg-info/requires.txt
src/selective_self_attention.egg-info/top_level.txt
src/trainers/__init__.py
src/trainers/callbacks.py
src/trainers/seeding.py
src/trainers/trainer.py
src/utils/__init__.py
src/utils/checkpoint.py
src/utils/config.py
src/utils/distributed.py
src/utils/logging.py
src/utils/math_utils.py
src/visualization/__init__.py
src/visualization/viz.py
tests/test_algorithms.py
tests/test_data.py
tests/test_eval.py
tests/test_losses.py
tests/test_models.py
tests/test_reproducibility.py
tests/test_ssa_basic.py