├── .github └── workflows │ └── python-publish.yml ├── .gitignore ├── LICENSE ├── README.md ├── data ├── README.md └── enwik8.gz ├── flash.png ├── flash_pytorch ├── __init__.py ├── autoregressive_wrapper.py └── flash_pytorch.py ├── setup.py └── train.py /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/FLASH-pytorch/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/FLASH-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/FLASH-pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/FLASH-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/FLASH-pytorch/HEAD/data/README.md -------------------------------------------------------------------------------- /data/enwik8.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/FLASH-pytorch/HEAD/data/enwik8.gz -------------------------------------------------------------------------------- /flash.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/FLASH-pytorch/HEAD/flash.png -------------------------------------------------------------------------------- /flash_pytorch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/FLASH-pytorch/HEAD/flash_pytorch/__init__.py -------------------------------------------------------------------------------- /flash_pytorch/autoregressive_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/FLASH-pytorch/HEAD/flash_pytorch/autoregressive_wrapper.py -------------------------------------------------------------------------------- /flash_pytorch/flash_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/FLASH-pytorch/HEAD/flash_pytorch/flash_pytorch.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/FLASH-pytorch/HEAD/setup.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucidrains/FLASH-pytorch/HEAD/train.py --------------------------------------------------------------------------------