torch>=1.10
numpy
tqdm
pyyaml
scipy
wandb
rich>=13.0.0
rich>=13.0.0

[all]
transformers
datasets
accelerate
bitsandbytes

[dev]
pytest>=7.0
pytest-cov
black
flake8
mypy
isort
