/pytorchviz

A small package to create visualizations of PyTorch execution graphs

Primary LanguageJupyter Notebook

Watchers