LICENSE
README.md
pyproject.toml
flexit/__init__.py
flexit/version.py
flexit/attention/__init__.py
flexit/attention/multi_head.py
flexit/attention/positional/__init__.py
flexit/attention/positional/absolute.py
flexit/attention/positional/alibi.py
flexit/attention/positional/base.py
flexit/attention/positional/relative.py
flexit/attention/positional/rotary.py
flexit/blocks/__init__.py
flexit/blocks/causal_decoder.py
flexit/blocks/cross_decoder.py
flexit/blocks/encoder.py
flexit/config/__init__.py
flexit/config/model_config.py
flexit/config/training_config.py
flexit/core/__init__.py
flexit/core/embeddings.py
flexit/core/feedforward.py
flexit/core/generator.py
flexit/core/normalization.py
flexit/factory/__init__.py
flexit/factory/model_factory.py
flexit/inference/__init__.py
flexit/inference/decoding.py
flexit/inference/sampling.py
flexit/layers/__init__.py
flexit/layers/causal_decoder_layer.py
flexit/layers/cross_decoder_layer.py
flexit/layers/encoder_layer.py
flexit/layers/sublayer.py
flexit/models/__init__.py
flexit/models/base.py
flexit/models/decoder_only.py
flexit/models/encoder_decoder.py
flexit/models/encoder_only.py
flexit/models/flexi.py
flexit/models/heads/__init__.py
flexit/models/heads/classification.py
flexit/models/heads/lm_head.py
flexit/models/heads/token_classification.py
flexit/training/__init__.py
flexit/training/batch.py
flexit/training/callbacks.py
flexit/training/loss.py
flexit/training/schedulers.py
flexit/training/state.py
flexit/training/trainer.py
flexit/utils/__init__.py
flexit/utils/helpers.py
flexit/utils/initialization.py
flexit/utils/masks.py
flexitransformers.egg-info/PKG-INFO
flexitransformers.egg-info/SOURCES.txt
flexitransformers.egg-info/dependency_links.txt
flexitransformers.egg-info/requires.txt
flexitransformers.egg-info/top_level.txt
tests/test_attention.py
tests/test_config.py
tests/test_inference.py
tests/test_models.py
tests/test_training.py
tests/test_version.py