A small package to create visualizations of PyTorch execution graphs
Primary LanguageJupyter Notebook