torch>=1.10.1
torchinfo>=1.6.0
numpy>=1.22.0
