torch>=2.0
numpy>=1.20
tqdm

[all]
transformers>=4.30
datasets
accelerate>=0.21.0
peft>=0.5.0
bitsandbytes>=0.41.0
scipy
wandb

[dev]
pytest
pytest-cov
ruff>=0.1

[llm]
bitsandbytes>=0.41.0
accelerate>=0.21.0
scipy

[peft]
peft>=0.5.0

[tracking]
wandb

[transformers]
transformers>=4.30
datasets
accelerate>=0.21.0
