NNsight and NDIF are introduced to enable the scientific study of representations and computations learned by large neural networks.
NNsight is an open-source system that extends PyTorch with deferred remote execution, and NDIF is a scalable inference service that executes NNsight requests.
The Intervention Graph architecture is developed to decouple experimental design from model runtime, enabling transparent and efficient access to the internals of deep neural networks.
The framework allows for a range of research methods on huge models and has been benchmarked with previous approaches.