accelerate
datasets==2.18.0
ml-collections
numpy
torch
tqdm
transformers==4.41.0

[all]
transformercompression[dev,experiment,finetune]

[dev]
mypy>=1.5.1
pre-commit
pylint>=3.0.0
pytest>=7.4.2

[experiment]
lm-eval==0.4.1
wandb

[finetune]
syne-tune[gpsearchers]==0.10.0
peft==0.6.2
