Installation

pip install hessian-eigenthings

Optional extras:

pip install "hessian-eigenthings[transformers]"        # HuggingFace Transformers helpers
pip install "hessian-eigenthings[transformer-lens]"    # TransformerLens helpers

To work on the library itself:

git clone https://github.com/noahgolmant/pytorch-hessian-eigenthings
cd pytorch-hessian-eigenthings
pip install -e ".[dev,docs]"