Comgra

A PyTorch library that extracts tensor data and visualizes it in a browser, helping researchers debug neural networks, track gradients and compare training runs.