VisualTorch
Architecture diagrams for PyTorch models, straight from a torch.nn.Module.
Layered, graph and LeNet-style views, skip connections, multi-input models and custom colors, made for papers, slides and docs. Installed about 37,000 times from PyPI.
