torchsummaryX
flyai
torch>=1.0.0
