Whittle is a library for Neural Architecture Search (NAS) aimed at compressing large language models (LLMs). The core idea is to treat the network as a super-network and select sub-networks that optimally balance performance and efficiency. Whittle provides the following functionalities:
Whittle supports and is tested for python 3.9 to 3.11.
You can install whittle with:
pip install whittle
Install whittle from source to get the most recent version:
git clone git@github.com:whittle-org/whittle.git
cd whittle
pip install -e .
To explore and understand different functionalities of whittle
checkout this colab notebook and examples/
We more than happy for any code contribution. If you are interested in contribution to whittle, please read our contribution guide.