├── .github └── workflows │ └── build.yml ├── .gitignore ├── CHANGELOG.md ├── LICENSE ├── README.md ├── dictionary_learning ├── __init__.py ├── activault_s3_buffer.py ├── buffer.py ├── config.py ├── dictionary.py ├── evaluation.py ├── grad_pursuit.py ├── interp.py ├── pytorch_buffer.py ├── trainers │ ├── __init__.py │ ├── batch_top_k.py │ ├── gated_anneal.py │ ├── gdm.py │ ├── jumprelu.py │ ├── matryoshka_batch_top_k.py │ ├── p_anneal.py │ ├── standard.py │ ├── top_k.py │ └── trainer.py ├── training.py └── utils.py ├── pretrained_dictionary_downloader.sh ├── pyproject.toml └── tests ├── test_end_to_end.py ├── test_pytorch_end_to_end.py └── unit └── test_dictionary.py /.github/workflows/build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/.github/workflows/build.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/README.md -------------------------------------------------------------------------------- /dictionary_learning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/__init__.py -------------------------------------------------------------------------------- /dictionary_learning/activault_s3_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/activault_s3_buffer.py -------------------------------------------------------------------------------- /dictionary_learning/buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/buffer.py -------------------------------------------------------------------------------- /dictionary_learning/config.py: -------------------------------------------------------------------------------- 1 | # debugging flag for use in other scripts 2 | DEBUG = False -------------------------------------------------------------------------------- /dictionary_learning/dictionary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/dictionary.py -------------------------------------------------------------------------------- /dictionary_learning/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/evaluation.py -------------------------------------------------------------------------------- /dictionary_learning/grad_pursuit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/grad_pursuit.py -------------------------------------------------------------------------------- /dictionary_learning/interp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/interp.py -------------------------------------------------------------------------------- /dictionary_learning/pytorch_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/pytorch_buffer.py -------------------------------------------------------------------------------- /dictionary_learning/trainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/trainers/__init__.py -------------------------------------------------------------------------------- /dictionary_learning/trainers/batch_top_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/trainers/batch_top_k.py -------------------------------------------------------------------------------- /dictionary_learning/trainers/gated_anneal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/trainers/gated_anneal.py -------------------------------------------------------------------------------- /dictionary_learning/trainers/gdm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/trainers/gdm.py -------------------------------------------------------------------------------- /dictionary_learning/trainers/jumprelu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/trainers/jumprelu.py -------------------------------------------------------------------------------- /dictionary_learning/trainers/matryoshka_batch_top_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/trainers/matryoshka_batch_top_k.py -------------------------------------------------------------------------------- /dictionary_learning/trainers/p_anneal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/trainers/p_anneal.py -------------------------------------------------------------------------------- /dictionary_learning/trainers/standard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/trainers/standard.py -------------------------------------------------------------------------------- /dictionary_learning/trainers/top_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/trainers/top_k.py -------------------------------------------------------------------------------- /dictionary_learning/trainers/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/trainers/trainer.py -------------------------------------------------------------------------------- /dictionary_learning/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/training.py -------------------------------------------------------------------------------- /dictionary_learning/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/dictionary_learning/utils.py -------------------------------------------------------------------------------- /pretrained_dictionary_downloader.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/pretrained_dictionary_downloader.sh -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/test_end_to_end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/tests/test_end_to_end.py -------------------------------------------------------------------------------- /tests/test_pytorch_end_to_end.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/tests/test_pytorch_end_to_end.py -------------------------------------------------------------------------------- /tests/unit/test_dictionary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/saprmarks/dictionary_learning/HEAD/tests/unit/test_dictionary.py --------------------------------------------------------------------------------