szagoruyko/pytorchviz
PyTorchViz renders PyTorch model computation graphs as visual diagrams for debugging and understanding model architecture.
Not currently ranked — collecting fresh signals.
star history
This package provides make_dot and related functions that generate Graphviz visualizations of PyTorch execution traces. It shows the forward pass computational graph and optionally reveals what autograd saves for backward pass debugging. It is used by ML practitioners to inspect layer connections and data flow within neural networks.
Frequently asked
- What is szagoruyko/pytorchviz?
- PyTorchViz renders PyTorch model computation graphs as visual diagrams for debugging and understanding model architecture.
- Is pytorchviz open source?
- Yes — szagoruyko/pytorchviz is open source, released under the MIT license.
- What language is pytorchviz written in?
- szagoruyko/pytorchviz is primarily written in Jupyter Notebook.
- How popular is pytorchviz?
- szagoruyko/pytorchviz has 3.5k stars on GitHub.
- Where can I find pytorchviz?
- szagoruyko/pytorchviz is on GitHub at https://github.com/szagoruyko/pytorchviz.