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
549 stars 55 forks source link

[Docs] Second-order Extension Example #320

Closed RobertDeibel closed 4 months ago

RobertDeibel commented 6 months ago

This PR