LICENSE
MANIFEST.in
README.md
pyproject.toml
src/stok/__init__.py
src/stok/train.py
src/stok.egg-info/PKG-INFO
src/stok.egg-info/SOURCES.txt
src/stok.egg-info/dependency_links.txt
src/stok.egg-info/entry_points.txt
src/stok.egg-info/requires.txt
src/stok.egg-info/top_level.txt
src/stok/checkpoints/codebook/base.pt
src/stok/checkpoints/codebook/lite.pt
src/stok/cli/cli.py
src/stok/cli/smoke_test.py
src/stok/cli/train.py
src/stok/configs/config.yaml
src/stok/configs/data/base.yaml
src/stok/configs/model/arch.yaml
src/stok/configs/train/base.yaml
src/stok/data/collate.py
src/stok/data/dataset.py
src/stok/models/attention.py
src/stok/models/blocks.py
src/stok/models/decoder.py
src/stok/models/encoder.py
src/stok/models/gcpnet.py
src/stok/models/head.py
src/stok/models/mlp.py
src/stok/models/rope.py
src/stok/models/stok.py
src/stok/utils/codebook.py
src/stok/utils/featurizer.py
src/stok/utils/gcp.py
src/stok/utils/geometry.py
src/stok/utils/losses.py
src/stok/utils/masking.py
src/stok/utils/pdb.py
src/stok/utils/sampling.py
src/stok/utils/tokenizer.py