accelerate>=0.17.1
attrs>=22.1.0
cattrs>=22.2.0
datasets
deepspeed>=0.8.1
einops>=0.4.1
numpy>=1.23.2
torchtyping
transformers>=4.27.1
tqdm
rich
wandb>=0.13.5
ray>=2.4.0
tabulate>=0.9.0
networkx
tritonclient

[bnb]
bitsandbytes
scipy

[dev]
black
hypothesis
isort
flake8
pre-commit
pytest
pytest-cov