google-deepmind/mishax

A JAX research toolkit for building, editing, and visualizing neural networks.

Unified access to Large Language Model modules using NNsight

Repo for vLLM Hook, an vLLM plug-in for programming internal states of models deployed on vLLM

The nnsight package enables interpreting and manipulating the internals of deep learned models.

Model interpretability and understanding for PyTorch