[ADD] Application of `bias_jac_t` for a subset of samples

f-dangel / backpack

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

https://backpack.pt/

MIT License

561 stars 55 forks source link

Closed f-dangel closed 3 years ago

f-dangel commented 3 years ago

Prepares the core functionality to support #12.