LICENSE
README.md
pyproject.toml
test/test_api.py
test/test_attention.py
test/test_checkpoint_loading.py
test/test_embedding.py
test/test_extract.py
test/test_flops.py
test/test_linear.py
test/test_lmhead.py
test/test_mag.py
test/test_mlp.py
test/test_model.py
test/test_norms.py
test/test_parameters.py
test/test_pretrain_super_network.py
test/test_search.py
test/test_training_strategies.py
test/test_transformer_block.py
whittle/__init__.py
whittle/__version__.py
whittle/pretrain_super_network.py
whittle.egg-info/PKG-INFO
whittle.egg-info/SOURCES.txt
whittle.egg-info/dependency_links.txt
whittle.egg-info/requires.txt
whittle.egg-info/top_level.txt
whittle/eval/__init__.py
whittle/eval/utils.py
whittle/eval/whittle_llms.py
whittle/loss/__init__.py
whittle/loss/kd_loss.py
whittle/metrics/__init__.py
whittle/metrics/flops.py
whittle/metrics/latency.py
whittle/metrics/mag.py
whittle/metrics/parameters.py
whittle/models/__init__.py
whittle/models/gpt/__init__.py
whittle/models/gpt/extract.py
whittle/models/gpt/model.py
whittle/models/gpt/utils.py
whittle/models/gpt/blocks/__init__.py
whittle/models/gpt/blocks/causal_self_attention.py
whittle/models/gpt/blocks/mlp.py
whittle/models/gpt/blocks/transformer_block.py
whittle/modules/__init__.py
whittle/modules/embedding.py
whittle/modules/layernorm.py
whittle/modules/linear.py
whittle/modules/rmsnorm.py
whittle/sampling/__init__.py
whittle/sampling/random_sampler.py
whittle/search/__init__.py
whittle/search/ask_tell_scheduler.py
whittle/search/baselines.py
whittle/search/local_search.py
whittle/search/multi_objective.py
whittle/search/search.py
whittle/training_strategies/__init__.py
whittle/training_strategies/ats.py
whittle/training_strategies/base_strategy.py
whittle/training_strategies/random.py
whittle/training_strategies/random_linear.py
whittle/training_strategies/sandwich.py
whittle/training_strategies/standard.py
whittle/tutorials/__init__.py
whittle/tutorials/gpt_utils.py