The newest version of Comgra is out! Thanks to everyone for all your feedback!

Comgra (computation graph analysis) is a library for pytorch to visualize and explore your computation graph in detail. The GUI allows you to look at your data from many different angles, to view examples, check summary statistics, investigate gradients, trace bugs through the network step-by-step, and more.

Use Tensorboard to get an overview of your model, and use Comgra for a deep-dive.

My goal is to make this the go-to library used both by novices who want to understand what’s going on under the hood, and by researchers in neural architecture design who need to know all the details.