torch
transformers
accelerate
numpy
typer
rich
pyyaml
sentencepiece
tiktoken
protobuf

[dev]
pytest
pytest-cov

[lm-eval]
lm-eval>=0.4.0
jieba
fuzzywuzzy
rouge

[viz]
matplotlib
plotly
scipy
