LICENSE
README.md
pyproject.toml
test/test_api.py
test/test_attention.py
test/test_checkpoint_loading.py
test/test_cli.py
test/test_distill_loss.py
test/test_embedding.py
test/test_evaluate.py
test/test_extract.py
test/test_flops.py
test/test_full_finetune.py
test/test_linear.py
test/test_lmhead.py
test/test_lora_attention.py
test/test_lora_block.py
test/test_lora_embedding.py
test/test_lora_gpt.py
test/test_lora_linear.py
test/test_lora_mlp.py
test/test_lora_qkv_linear.py
test/test_mag.py
test/test_mlp.py
test/test_model.py
test/test_model_checkpoint.py
test/test_norms.py
test/test_parameters.py
test/test_pretrain_super_network.py
test/test_prune.py
test/test_pruners.py
test/test_search.py
test/test_search_sub_network.py
test/test_training_strategies.py
test/test_transformer_block.py
whittle/__init__.py
whittle/__main__.py
whittle/__version__.py
whittle/args.py
whittle/convert_to_litgpt.py
whittle/distill.py
whittle/evaluate_network.py
whittle/full_finetune.py
whittle/lora.py
whittle/pretrain_super_network.py
whittle/prune.py
whittle/search_sub_networks.py
whittle.egg-info/PKG-INFO
whittle.egg-info/SOURCES.txt
whittle.egg-info/dependency_links.txt
whittle.egg-info/entry_points.txt
whittle.egg-info/requires.txt
whittle.egg-info/top_level.txt
whittle/data/__init__.py
whittle/data/llamamini.py
whittle/eval/__init__.py
whittle/eval/utils.py
whittle/eval/whittle_llms.py
whittle/lora_model/__init__.py
whittle/lora_model/config.py
whittle/lora_model/lora_attention.py
whittle/lora_model/lora_block.py
whittle/lora_model/lora_embedding.py
whittle/lora_model/lora_gpt.py
whittle/lora_model/lora_linear.py
whittle/lora_model/lora_mlps.py
whittle/lora_model/lora_qkv_linear.py
whittle/lora_model/merge.py
whittle/loss/__init__.py
whittle/loss/kd_loss.py
whittle/loss/losses.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/checkpoint.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/pruning/__init__.py
whittle/pruning/pruners/__init__.py
whittle/pruning/pruners/base_pruner.py
whittle/pruning/pruners/magnitude.py
whittle/pruning/pruners/sparsegpt.py
whittle/pruning/pruners/wanda.py
whittle/pruning/utils/__init__.py
whittle/pruning/utils/catcher.py
whittle/pruning/utils/layerwrapper.py
whittle/pruning/utils/sparsegpt.py
whittle/sampling/__init__.py
whittle/sampling/grid_samplers.py
whittle/sampling/param_bins.py
whittle/sampling/random_sampler.py
whittle/sampling/samplers.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/search/search_spaces.py
whittle/search/stratified_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/training_strategies/strategies.py
whittle/tutorials/__init__.py
whittle/tutorials/gpt_utils.py