torch
transformers
accelerate
numpy
typer
rich
pyyaml

[dev]
pytest
pytest-cov

[viz]
matplotlib
plotly
