BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
549
stars
55
forks
source link
[ADD] Support `BCEWithLogitsLoss` in `SqrtGGN{Exact, MC}` #282
Closed
f-dangel closed 1 year ago