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/metrics.py
stackformer/py.typed
stackformer/trainer.py
stackformer/amp/__init__.py
stackformer/amp/scaler.py
stackformer/distributed/__init__.py
stackformer/distributed/ddp.py
stackformer/engine/__init__.py
stackformer/engine/checkpoint.py
stackformer/engine/engine.py
stackformer/engine/state.py
stackformer/engine/trainer.py
stackformer/logging/__init__.py
stackformer/logging/csv_logger.py
stackformer/logging/logger.py
stackformer/logging/metrics.py
stackformer/logging/tensorboard_logger.py
stackformer/logging/wandb_logger.py
stackformer/logging/wb_logger.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/Masking.py
stackformer/modules/Normalization.py
stackformer/modules/__init__.py
stackformer/modules/position_embedding.py
stackformer/modules/tokenizer.py
stackformer/optim/__init__.py
stackformer/optim/factories.py
stackformer/optim/loss_fn.py
stackformer/training/__init__.py
stackformer/training/loops.py
stackformer/utils/__init__.py
stackformer/utils/device.py
stackformer/utils/utils.py
stackformer/vision/__init__.py
stackformer/vision/segformer.py
stackformer/vision/vit.py
tests/test_distributed.py
tests/test_vision.py