torch
transformers
accelerate
numpy
typer
rich
pyyaml
sentencepiece
tiktoken
protobuf

[dev]
pytest
pytest-cov

[viz]
matplotlib
plotly
