johnmarktaylor91 / torchlens

git clone https://github.com/johnmarktaylor91/torchlens.git

Capture every activation and gradient of any PyTorch model — forward and backward — with automatic graph visualization, rich metadata, and live interventions. Works on any architecture, including dynamic and recurrent ones.