torch
transformers
accelerate
fastchat
shortuuid
tqdm
datasets
numpy
psutil
nvidia-ml-py
safetensors

[dev]
pytest
ruff
build
twine

[quant]
bitsandbytes
