A small package to create visualizations of PyTorch execution graphs - View it on GitHub