torch>=2.1
transformers>=4.40
accelerate
datasets
safetensors
numpy
tqdm

[dev]
pytest
ruff
build
twine

[eval]
lm-eval>=0.4

[plot]
matplotlib

[qtorch]
qtorch_plus==0.2.0
ninja
