├── .gitignore ├── README.md ├── requirements.txt └── src ├── barrel_rec.py ├── caterpillar.py ├── data.py ├── dumb_rec.py ├── grid.py ├── model.py ├── pscan.py ├── pscan_cumsum.py ├── pscan_fft.py ├── qkv_attention.py ├── rotary_embedding.py ├── train.py └── train_gpt2.py /.gitignore: -------------------------------------------------------------------------------- 1 | .venv 2 | *.pyc 3 | __pycache__ 4 | wandb 5 | runs 6 | outputs 7 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/barrel_rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/barrel_rec.py -------------------------------------------------------------------------------- /src/caterpillar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/caterpillar.py -------------------------------------------------------------------------------- /src/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/data.py -------------------------------------------------------------------------------- /src/dumb_rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/dumb_rec.py -------------------------------------------------------------------------------- /src/grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/grid.py -------------------------------------------------------------------------------- /src/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/model.py -------------------------------------------------------------------------------- /src/pscan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/pscan.py -------------------------------------------------------------------------------- /src/pscan_cumsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/pscan_cumsum.py -------------------------------------------------------------------------------- /src/pscan_fft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/pscan_fft.py -------------------------------------------------------------------------------- /src/qkv_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/qkv_attention.py -------------------------------------------------------------------------------- /src/rotary_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/rotary_embedding.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/train.py -------------------------------------------------------------------------------- /src/train_gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dvruette/barrel-rec-pytorch/HEAD/src/train_gpt2.py --------------------------------------------------------------------------------