torch
transformers
accelerate
fschat[llm_judge]
shortuuid
tqdm
datasets
numpy
psutil
nvidia-ml-py
safetensors

[dev]
pytest
ruff
build
twine

[plot]
matplotlib

[quant]
bitsandbytes
