├── .flake8 ├── .gitignore ├── LICENSE ├── README.md ├── adasplash ├── __init__.py ├── adasplash_block_mask.py ├── adasplash_no_block_mask.py └── triton_entmax.py ├── benchmark.png ├── pyproject.toml ├── requirements-dev.txt ├── requirements.txt ├── setup.py └── tests ├── test_adasplash.py ├── test_adasplash_no_block_mask.py └── test_triton_entmax.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/README.md -------------------------------------------------------------------------------- /adasplash/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/adasplash/__init__.py -------------------------------------------------------------------------------- /adasplash/adasplash_block_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/adasplash/adasplash_block_mask.py -------------------------------------------------------------------------------- /adasplash/adasplash_no_block_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/adasplash/adasplash_no_block_mask.py -------------------------------------------------------------------------------- /adasplash/triton_entmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/adasplash/triton_entmax.py -------------------------------------------------------------------------------- /benchmark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/benchmark.png -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- 1 | pytest 2 | black 3 | isort 4 | flake8 5 | entmax -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | torch 3 | triton>=3.1.0 -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_adasplash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/tests/test_adasplash.py -------------------------------------------------------------------------------- /tests/test_adasplash_no_block_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/tests/test_adasplash_no_block_mask.py -------------------------------------------------------------------------------- /tests/test_triton_entmax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/deep-spin/adasplash/HEAD/tests/test_triton_entmax.py --------------------------------------------------------------------------------