f-dangel

BackPACK - 基於 PyTorch 建構的反向傳播套件，能有效率地計算梯度以外的量。