GitHub repository: Gainward777/torchview
Description: torchview: visualize pytorch models