A small package to create visualizations of PyTorch execution graphs
conda install dnachun::torchviz