BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.
555
stars
55
forks
source link
[core] Implement manual sampled gradients of `BCEWithLogitsLoss`, use by default #278
Closed
f-dangel closed 1 year ago