LICENSE
README.md
pyproject.toml
stackformer/__init__.py
stackformer/config.py
stackformer/generate.py
stackformer/metrics.py
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/cache/__init__.py
stackformer/cache/paged.py
stackformer/cache/static.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/language/__init__.py
stackformer/language/decoder.py
stackformer/language/encoder_decoder.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/__init__.py
stackformer/models/bert.py
stackformer/models/gemma.py
stackformer/models/gpt.py
stackformer/models/llama.py
stackformer/models/roberta.py
stackformer/models/transformer.py
stackformer/modules/Attention.py
stackformer/modules/Feed_forward.py
stackformer/modules/Normalization.py
stackformer/modules/__init__.py
stackformer/modules/attention_engine.py
stackformer/modules/layer.py
stackformer/modules/masks.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/attn_utils.py
stackformer/utils/cache.py
stackformer/utils/device.py
stackformer/utils/ff_utils.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