torch>=2.0
numpy>=1.20
pydantic>=2.0
tqdm>=4.64
pyyaml>=6.0

[data]
pandas
pyarrow>=10.0
fastparquet

[dev]
pytest
pytest-cov
tensorboard>=2.21
ruff>=0.15
mypy>=2.0
pre-commit>=4.0

[eval]
lm-eval>=0.4
transformers>=4.30

[hf]
transformers>=4.30

[industrial]
megatron-core<0.17,>=0.16
transformers>=4.44
trl>=1.0
peft>=0.13
accelerate>=0.30
datasets>=3.0
tokenizers>=0.19
deepspeed<0.17,>=0.16

[quality]
openai>=1.0
fasttext-wheel>=0.9

[serve]
fastapi>=0.110
uvicorn>=0.29
transformers>=4.30

[tools]
zhconv
safetensors>=0.4

[train]
tensorboard>=2.21
