LICENSE
README.md
pyproject.toml
stackformer/__init__.py
stackformer/config.py
stackformer/generate.py
stackformer/metrics.py
stackformer/py.typed
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/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/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