Rare find

torchlens. 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.

github.com/johnmarktaylor91/torchlens

Problem

Audience

Updates

No founder updates yet. Check back soon.