a visualization tool to investigate bottlenecks of the computational graph in PyTorch - View it on GitHub