torchinfo-viz turns any PyTorch model into an interactive view: you can see each layer, how layers connect (including skip connections), the tensor shapes, and which layers hold the most parameters.
pip install torchinfo-viz
built on torchinfo (MIT). feedback welcome!