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