It is a Keras style model.summary() implementation for PyTorch
conda install conda-forge::pytorch-model-summary