LICENSE
README.md
pyproject.toml
Stackformer.egg-info/PKG-INFO
Stackformer.egg-info/SOURCES.txt
Stackformer.egg-info/dependency_links.txt
Stackformer.egg-info/requires.txt
Stackformer.egg-info/top_level.txt
stackformer/__init__.py
stackformer/generate.py
stackformer/py.typed
stackformer/trainer.py
stackformer/models/Google.py
stackformer/models/Meta.py
stackformer/models/OpenAI.py
stackformer/models/Transformer.py
stackformer/models/__init__.py
stackformer/modules/Attention.py
stackformer/modules/Feed_forward.py
stackformer/modules/Normalization.py
stackformer/modules/__init__.py
stackformer/modules/mask.py
stackformer/modules/position_embedding.py
stackformer/modules/tokenizer.py
stackformer/vision/__init__.py
stackformer/vision/segformer.py
stackformer/vision/vit.py
tests/test_attention.py
tests/test_ff.py
tests/test_models.py
tests/test_normalization.py
tests/test_position_embedding.py
tests/test_vision.py