numpy>=1.24

[gpu]
cupy-cuda12x>=13.0

[train]
torch>=2.1
datasets
