/pytorchviz

A small package to create visualizations of PyTorch execution graphs

Primary LanguageJupyter NotebookMIT LicenseMIT

Watchers

No one’s watching this repository yet.