/backpack

BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.

Primary LanguagePythonMIT LicenseMIT

Watchers