numpy>=1.24

[train]
torch>=2.1
datasets
