GitHub repository: sytelus/torchstat
Description: Model analyzer in PyTorch