/pytorchviz

A small package to create visualizations of PyTorch execution graphs

Primary LanguageJupyter NotebookMIT LicenseMIT

Watchers