torch>=1.9.0
torchinfo
numpy
