numpy

[train]
torch>=2.4
pyyaml
