/backpack

BACKpropagation PACKage - A backpack for PyTorch that extends the backward pass of feedforward networks to compute quantities beyond the gradient

Primary LanguagePythonMIT LicenseMIT

BACKpropagation PACKage - a backpack for PyTorch

A backpack for PyTorch that extends the backward pass of feedforward networks to compute quantities beyond the gradient.

Check out the examples on how to use the code.

Installation

pip install backpack-for-pytorch

Alternative: Clone the repository and run setup.py

git clone https://github.com/f-dangel/backpack.git ~/backpack
cd ~/backpack
python setup.py install