numpy
torch
matplotlib
tqdm
huggingface_hub
pandas

[dev]
pytest

[train]
wandb
pyyaml
