pytorch-lightning
torch
torchinfo
