torch==2.13.0
typing_extensions>=4.0.0
tqdm

[dev]
mypy==1.20.2
pylint==2.15.7
pycodestyle==2.14.0
black==26.5.1
pytest

[huggingface]
transformers>=5.10.0
datasets

[triton]
triton>=2.0.0
