torch>=2.5.1

[test]
matplotlib
numpy
tqdm
