torch>=1.8.1

numpy>=1.19.5