What AI says about TorchLens

TorchLens is a Python package that extracts activations from every intermediate operation in a PyTorch model with a single line of code and automatically visualizes the computational graph. It also supports intervention-ready logging, allowing users to fork, edit, and replay forward passes for causal analysis.